#175-【线段树与树状数组】砍树

Description

在一个凉爽的夏夜，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不是π）。

 

Input

第一行，一个整数n。
第二行，共n个整数，表示每棵树的高度
第三行，一个整数m，表示共m个区间。
以下m行，每个区间[x,y]的左右端点x、y。

 

Output

共m行，每行一个数，表示每个区间的木材量。结果精确到小数点后两位。

Sample Input

5 
1 2 3 4 5 
2 
1 4 
2 4 

Sample Output

31.40 
21.98

HINT

【备注】
对于30%的数据，有n<=5000，m<=5000；
对于100%的数据，有n<=200000，m<=200000;
【样例解释】
第一次砍[1，4]的树后，森林变为：1 0 3 4 5

线段树/树状数组模板题

树状数组解法:

#include <iostream>

#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 <iostream>

#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;
}