线段树---HDU1166敌兵布阵

  1 #include <stdio.h>
  2 #include <math.h>
  3 
  4 const int MAX = 50010 * 4;
  5 int segment[MAX];//存放线段树，因为类似完全二叉树， 所以可以用数组来表示
  6 //更新root节点的值，即兵营里的人数
  7 void pushUp(int root, int add_num)
  8 {
  9     segment[root] = segment[root * 2] + segment[root * 2 + 1];
 10 }
 11 //建树，只需要两个点，一个起点，一个终点
 12 void buildTree(int root, int left, int right)
 13 {
 14     if(left == right)
 15     {
 16         //输入兵营里的人数
 17         scanf("%d", &segment[root]);
 18         return;
 19     }
 20     int mid = (left + right) / 2;
 21     buildTree(root * 2, left, mid);
 22     buildTree(root * 2 + 1, mid + 1, right);
 23     //调整它的上面节点的值
 24     pushUp(root);
 25 }
 26 /*更新最下面节点的值，而且要更新以上给他有关联的节点的值, root代表根节点，
 27 pos代表更新的位置， add_num 代表增加的值，如果是负数，说明是减少的，left和right
 28 分别为当前节点区间的左右端点*/
 29 void update(int root, int pos, int add_num, int left, int right)
 30 {
 31     if (left == right)
 32     {
 33         segment[root] += add_num;
 34         return;
 35     }
 36     int mid = (left + right) / 2;
 37     if (pos <= mid)
 38         update(root * 2, pos, add_num, left, mid);
 39     else
 40         update(root * 2 + 1, pos, add_num, mid + 1, right);
 41     //向上调整
 42     pushUp(root);
 43 }
 44 //获取指定区间内的总数
 45 int getSum(int root, int left, int right, int L, int R)
 46 {
 47     if(left == L && right == R)
 48     {
 49         return segment[root];
 50     }
 51     int mid = (L + R) / 2;
 52     int res = 0;
 53     //如果在当前节点的右半个区间内
 54     if(left > mid)
 55     {
 56         res += getSum(root * 2 + 1, left, right, mid + 1, R);
 57     }
 58     //如果在当前节点的左半个区间内
 59     else if(right <= mid)
 60     {
 61         res += getSum(root * 2, left, right, L, mid);
 62     }
 63     //一个在左边，一个在右边
 64     else
 65     {
 66         res += getSum(root * 2, left, mid, L, mid);
 67         res += getSum(root * 2 + 1, mid + 1, right, mid + 1, R);
 68     }
 69     return res;
 70 }
 71 
 72 int main()
 73 {
 74     int T;
 75     scanf("%d", &T);
 76     for(int kase = 1; kase <= T; kase++)
 77     {
 78 
 79         int n;
 80         scanf("%d", &n);
 81         buildTree(1, 1, n);//建树，同时输入节点的值，也就是兵营的人数
 82         char op[10];
 83         int t1, t2;
 84         printf("Case %d:\n", kase);
 85         while(scanf("%s", op))
 86         {
 87             if(op[0] == ‘E‘)
 88                 break;
 89             scanf("%d %d", &t1, &t2);
 90             if(op[0] == ‘A‘)
 91             {
 92                 update(1, t1, t2, 1, n);
 93             }
 94             else if(op[0] == ‘S‘)
 95             {
 96                 update(1, t1, -t2, 1, n);
 97             }
 98             else
 99             {
100                 printf("%d\n", getSum(1, t1, t2, 1, n));
101             }
102         }
103     }
104     return 0;
105 }