hdu 1166 敌兵布阵(线段树)

http://acm.hdu.edu.cn/showproblem.php?pid=1166

人生第一道线段树题目，一直觉得线段树好高端。。该题应该是线段树最简单的问题，单点增减，包括建树，查询，更新。

#include <stdio.h>
#include <string.h>
const int maxn = 50005;

//线段树节点定义，表示某一线段的左端点，右端点，以及该区间的人数之和
struct line
{
	int l;
	int r;
	int value;
}tree[4*maxn];

int n,a[maxn];

//建树，v是当前线段树的编号，l,r表示对应线段树的左右端点
void build(int v, int l, int r)
{
	tree[v].l = l;
	tree[v].r = r;
	if(l == r)
	{
		tree[v].value = a[r];
		return;
	}

	int mid = (l+r)>>1;
	build(v*2,l,mid);
	build(v*2+1,mid+1,r);
	tree[v].value = tree[v*2].value + tree[v*2+1].value;
}

//查询左右端点是l,r的线段树的value，v是待查询节点
int query(int v, int l, int r)
{
	if(tree[v].l == l && tree[v].r == r)
	{
		return tree[v].value;
	}

	int mid = (tree[v].l+tree[v].r)>>1;
	if(r <= mid)
		return query(v*2,l,r);
	else if(l > mid)
		return query(v*2+1,l,r);
	else return (query(v*2,l,mid) + query(v*2+1,mid+1,r));
}

//单点更新，更新端点是pos的节点，value增加add
void update(int v,int pos,int add)
{
	if(tree[v].l == tree[v].r && tree[v].r == pos)
	{
		tree[v].value += add;
		return;
	}

	int mid = (tree[v].l + tree[v].r)>>1;
	if(pos <= mid)
		update(v*2,pos,add);
	else update(v*2+1,pos,add);
	tree[v].value = tree[v*2].value + tree[v*2+1].value;//儿子节点值改变，父亲节点值也要改变

}

int main()
{
	int test;
	char str[10];
	scanf("%d",&test);
	for(int item = 1; item <= test; item++)
	{
		scanf("%d",&n);
		for(int i = 1; i <= n; i++)
			scanf("%d",&a[i]);

		build(1,1,n);
		printf("Case %d:\n",item);

		int l,r,ans;
		int pos,add;
		while(scanf("%s",str) && str[0] != 'E')
		{
			if(str[0] == 'Q')
			{
				scanf("%d %d",&l,&r);
				ans = query(1,l,r);
				printf("%d\n",ans);
			}

			else if(str[0] == 'A')
			{
				scanf("%d %d",&pos,&add);
				update(1,pos,add);
			}
			else if(str[0] == 'S')
			{
				scanf("%d %d",&pos,&add);
				update(1,pos,-add);
			}
		}

	}
	return 0;
}