在一个凉爽的夏夜,xth 和 rabbit 来到花园里砍树。为啥米要砍树呢?是这样滴,
小菜儿的儿子窄森要出生了。Xth这个做伯伯的自然要做点什么。于是他决定带着
rabbit 去收集一些木材,给窄森做一个婴儿车……(xth 早就梦想着要天天打菜儿
他儿窄森的小 pp,到时候在婴儿车里安装一个电子遥控手臂,轻轻一按,啪啪
啪……“乌卡卡——”xth 邪恶滴笑了,“不要告诉 rabbit,她会说我缺德的……”
xth 如是说)。
花园里共有n棵树。为了花园的整体形象,rabbit 要求 xth只能在m个区域砍伐,我
们可以将这m个区域看成m个区间,树的间距相等,都是1,我们将每个区间设为
[x, y]。那么长度为k的区间中就有k棵树。树木的高度不等。现在 xth 想测量一下,
每个区间树木砍伐后所得的木材量是多少,而且每次测量后他都会砍下标号为
(x+y)/2
的那棵作为纪念。以方便他安排人手。(同一个区间的树木可以重复砍伐,我们认
为被砍过的树木高度为0)
每棵树的木材量=树的高度∗ 3.14(注意是3.14不是π)。
第一行,一个整数n。
第二行,共n个整数,表示每棵树的高度
第三行,一个整数m,表示共m个区间。
以下m行,每个区间[x,y]的左右端点x、y。
共m行,每行一个数,表示每个区间的木材量。结果精确到小数点后两位。
5
1 2 3 4 5
2
1 4
2 4
31.40
21.98
【备注】
对于30%的数据,有n<=5000,m<=5000;
对于100%的数据,有n<=200000,m<=200000;
【样例解释】
第一次砍[1,4]的树后,森林变为:1 0 3 4 5
线段树/树状数组模板题
树状数组解法:
#include
#define SIZE 300010
#define lowbit(x) (x & (-x)) // 询问二进制最后的1代表的权
using namespace std;
int a[SIZE], c[SIZE], n;
void update(int x, int y) // 更新
{
while (x <= n)
{
c[x] += y; // 与其关联的一路加上去
x += lowbit(x);
}
return;
}
int query(int x) // 询问!
{
int res = 0;
while (x > 0)
{
res += c[x];
x -= lowbit(x); // 与上面的不同,询问的下标是要减下去
}
return res;
}
int main(void)
{
int m, i, x, y;
scanf("%d", &n);
for (i = 1; i <= n; ++i)
{
scanf("%d", &a[i]);
update(i, a[i]); // 初始化
}
scanf("%d", &m);
while (m--)
{
scanf("%d%d", &x, &y);
printf("%.2f\n", (query(y) - query(x - 1)) * 3.14); // 询问区间得要相减
update((x + y) >> 1, -a[(x+y)>>1]); // 更新 : 树被砍, 归0
a[(x+y)>>1] = 0;
}
return 0;
}
线段树解法:
#include
#define SIZE 800010
using namespace std;
int res[SIZE];
// 详细解释自行去看更加模板的 #174
void buildtree(int pos, int l, int r) // 建树
{
int mid;
if (l == r)
{
scanf("%d", &res[pos]);
return;
}
mid = l + r >> 1;
buildtree(pos << 1, l, mid);
buildtree((pos << 1) + 1, mid + 1, r);
res[pos] = res[pos<<1] + res[(pos<<1)+1];
return;
}
void update(int pos, int l, int r, int x, int y) // 更新一个节点
{
int mid;
if (l == r)
{
res[pos] = y;
return;
}
mid = l + r >> 1;
if (x <= mid)
{
update(pos << 1, l, mid, x, y);
}
else
{
update((pos << 1) + 1, mid + 1, r, x, y);
}
res[pos] = res[pos<<1] + res[(pos<<1)+1];
return;
}
int query(int pos, int l, int r, int x, int y) // 询问区间 [x, y] 中的高度总和.
{
int t1, t2, mid;
if ((r < x) || (l > y))
{
return 0;
}
if ((x <= l) && (y >= r))
{
return res[pos];
}
mid = l + r >> 1;
t1 = t2 = 0;
if (x <= mid)
{
t1 = query(pos << 1, l, mid, x, y);
}
if (y > mid)
{
t2 = query((pos << 1) + 1, mid + 1, r, x, y);
}
return t1 + t2;
}
int main(void)
{
int n, m, x, y;
scanf("%d", &n);
buildtree(1, 1, n);
scanf("%d", &m);
while (m--)
{
scanf("%d%d", &x, &y); // 输入区间!
printf("%.2f\n", query(1, 1, n, x, y) * 3.14); // 什么?查询区间的综合 * 3.14 !
update(1, 1, n, x + y >> 1, 0); // 区间中间有树? 不顺眼! 砍掉!
}
return 0;
}