线段树练习

题目描述 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 100010int 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;}