【原创】hdu 1166 敌兵布阵(线段树→单点更新，区间查询)

 1 #include<cstdio>
 2 #include<string>
 3 #include<iostream>
 4 #define N 50003
 5 using namespace std;
 6 int num[50003];
 7 char s[5];
 8 struct nod
 9 {
10     int data,l,r;
11 }tree[N*4];//树型结构的要花费更多的空间，这里要*4
12 
13 void push_up(int i)//通过i节点的孩子节点更新i节点
14 {
15     tree[i].data = tree[i*2].data + tree[i*2+1].data;
16 }
17 
18 void Build_tree(int i,int l,int r)
19 {
20     tree[i].l = l;
21     tree[i].r = r;
22     if(l == r){
23         tree[i].data = num[l];
24         return ;
25     }
26     int mid = (l + r)/2;
27     Build_tree(i*2,l,mid);
28     Build_tree(i*2+1,mid+1,r);
29     push_up(i);
30 }
31 
32 void update(int i,int k,int v)
33 {
34     if(tree[i].l==k&&tree[i].r==k){//单点更新只要找到特定的k节点就算找到，区间更新要找到一个区间→看我另一篇文章吧
35         tree[i].data += v;//注意题目意思是增加或减少V个人,所以是+=
36         return;
37     }
38     int mid = (tree[i].l + tree[i].r)/2;
39     if(k <= mid)
40         update(i*2,k,v);
41     else
42         update(i*2+1,k,v);
43     push_up(i);
44 }
45 
46 int query(int i,int l,int r)
47 {
48     if(l<=tree[i].l&&tree[i].r<=r){
49         return tree[i].data;
50     }
51     int mid = (tree[i].l + tree[i].r)/2;
52     if(r <= mid)　　　　　　//这里看不懂的看我另一篇文章http://www.cnblogs.com/liwenchi/p/5761257.html
53         return query(i*2,l,r);
54     if(l > mid)
55         return query(i*2+1,l,r);
56     return query(i*2,l,r)+query(i*2+1,l,r);
57 }
58 
59 int main()
60 {
61     int noc,ug;    //ug这名字随便起的...case数等于ug-noc,noc也是随便起的...number of case
62     scanf("%d",&noc);
63     ug = noc;
64     while(noc--)
65     {
66         int n,l,r,i,j;
67         scanf("%d",&n);
68         for(int i=1;i<=n;i++)
69             scanf("%d",&num[i]);
70         Build_tree(1,1,n);
71         printf("Case %d:\n",ug-noc);
72         while(scanf("%s",s))
73         {
74             if(s[0] == ‘E‘) break;
75             if(s[0] == ‘Q‘){
76                 scanf("%d%d",&l,&r);
77                 printf("%d\n",query(1,l,r));
78             }
79             else if(s[0] == ‘A‘){
80                 scanf("%d%d",&i,&j);
81                 update(1,i,j);
82             }
83             else{
84                 scanf("%d%d",&i,&j);//减人数的话把j取负再更新就OK了
85                 update(1,i,-j);
86             }
87         }
88     }
89 }