线段树练习

题目描述 Description

一行N个方格，开始每个格子里都有一个整数。现在动态地提出一些问题和修改：提问的形式是求某一个特定的子区间[a,b]中所有元素的和；修改的规则是指定某一个格子x，加上或者减去一个特定的值A。现在要求你能对每个提问作出正确的回答。1≤N<100000，,提问和修改的总数m<10000条。

输入描述 Input Description

输入文件第一行为一个整数N，接下来是n行n个整数，表示格子中原来的整数。接下一个正整数m，再接下来有m行，表示m个询问，第一个整数表示询问代号，询问代号1表示增加，后面的两个数x和A表示给位置X上的数值增加A，询问代号2表示区间求和，后面两个整数表示a和b，表示要求[a,b]之间的区间和。

输出描述 Output Description

共m行，每个整数

样例输入 Sample Input

6

4 

5 

6 

2 

1 

3

4

1 3 5

2 1 4

1 1 9

2 2 6

样例输出 Sample Output

22

22

#include<stdio.h>
#include<string.h>
#define MAX 100010
int data[MAX];
struct tree
{
	int left,right,digit;
};
tree segment_tree[4*MAX];
void build(int l,int r,int node)
{
	segment_tree[node].left=l;
	segment_tree[node].right=r;
	segment_tree[node].digit=data[l];
	if(l==r) return;
	int mid=(l+r)/2;
	build(l,mid,2*node+1);
	build(mid+1,r,2*node+2);
	segment_tree[node].digit=(segment_tree[2*node+1].digit+segment_tree[2*node+2].digit);
}
void insert(int pos,int update,int node)
{
	if(segment_tree[node].left==segment_tree[node].right)
	{
		segment_tree[node].digit+=update;
		return;
	}
	int mid=(segment_tree[node].left+segment_tree[node].right)/2;
	if(pos<=mid)
		insert(pos,update,2*node+1);
	else insert(pos,update,2*node+2);
	segment_tree[node].digit=(segment_tree[2*node+1].digit+segment_tree[2*node+2].digit);
}
int query(int l,int r,int node)
{
	if(segment_tree[node].left==l&&segment_tree[node].right==r)
		return segment_tree[node].digit;
	int mid=(segment_tree[node].left+segment_tree[node].right)/2;
	if(l<=mid)
	{
		if(r<=mid) return query(l,r,2*node+1);
		else return query(l,mid,2*node+1)+query(mid+1,r,2*node+2);
	}
	else query(l,r,2*node+2);
}
int main()
{
	//freopen("b.txt","r",stdin);
	int n,m,i,command,pos,update,l,r;
	while(scanf("%d",&n)==1)
	{
		for(i=1;i<=n;i++)
			scanf("%d",&data[i]);
		build(1,n,0);
		scanf("%d",&m);
		while(m--)
		{
			scanf("%d",&command);
			if(command==1)
			{
				scanf("%d %d",&pos,&update);
				insert(pos,update,0);
			}
			else
			{
				scanf("%d %d",&l,&r);
				printf("%d\n",query(l,r,0));
			}
		}
	}
	return 0;
}