线段树入门（1）

题目描述 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 数据范围及提示 Data Size & Hint 1≤N<100000， m<10000 。

地址：http://wikioi.com/problem/1080/

思路（见代码注释）

代码

[cpp]  view plain copy print ?

1. # include <cstdio>  
2. # define N 500000         
3.                                        // 利用二叉树的父子节点关系来存储，因此必须保证数组开的足够大  
4. struct node{  
5.     int l,r,v;              // left、right、value ；左右孩子由二叉树性质计算。  
6. };  
7. node st[N];  
8. int a[N];  
9.   
10. void build(int v,int l,int r){  
11.     st[v].l = l;  
12.     st[v].r = r;  
13.     if (l == r) {                   // 是叶子，直接赋值，跳出  
14.         st[v].v=a[l];  
15.         return ;  
16.     }  
17.     int mid = (l+r)/2;          // 不是叶子，往下扩展  
18.     build(2*v,l,mid);  
19.     build(2*v+1,mid+1,r);  
20.     st[v].v=st[v*2].v+st[v*2+1].v;  // 扩展出孩子后才能计算value域  
21. }  
22.   
23. void insert(int v,int w,int p){         // 在以v为根的树中，寻找节点w，并把节点w的值加上p 。 格外注意 某些编程语言中insert不能作为标识符  
24.     if (w>=st[v].l && w<=st[v].r){      
25.         st[v].v += p;           // 父子之间为统治关系，凡是经过的节点都要更新value  
26.     }  
27.     if (st[v].l==st[v].r) return;       // 找到终点，跳出  
28.     int mid = (st[v].l+st[v].r)/2;    
29.     if (w<=mid) {                // 判断左右孩子  
30.         insert(v*2,w,p);              
31.     }else {  
32.         insert(v*2+1,w,p);  
33.     }  
34. }  
35.   
36. int getsum(int v,int l,int r){          // 返回以v为根节点的树中，区间 [l,r] 的value域  
37.     if (st[v].l==l && st[v].r==r){  
38.         return st[v].v;         // 找到，返回  
39.     }  
40.     int mid = (st[v].l+st[v].r)/2;  
41.     if (r<=mid) {  
42.         return getsum(v*2,l,r);     // 区间 [l,r] 在v的左枝上  
43.     }else if (l>mid) {  
44.         return getsum(v*2+1,l,r);   // 右枝  
45.     }else {                 // 兵分两路  
46.         return getsum(v*2,l,mid) + getsum(v*2+1,mid+1,r);  
47.     }  
48. }  
49.   
50. int main(void){  
51.     freopen("1080.in","r",stdin);  
52.     int n;  
53.     scanf("%d",&n);  
54.     for (int i(1);i<=n;i++) {  
55.         scanf("%d",&a[i]);  
56.     }  
57.     build(1,1,n);  
58.     int m,t,a,b;  
59.     for (scanf("%d",&m);m;m--){  
60.         scanf("%d%d%d",&t,&a,&b);  
61.         if (t & 1) {  
62.             insert(1,a,b);  
63.         }else {  
64.             printf("%d\n",getsum(1,a,b));  
65.         }  
66.     }  
67.     return 0;  
68. }