更新节点区间求和- HDU 1166 敌兵…

线段树第一题。

hdu1166 敌兵布阵

题目大意：给定 N 个工兵营地，编号为 1--N，每个营地有相应的人数。现在对连续的工兵营地进行操作：

(1)Add i j,i和j为正整数,表示第i个营地增加j个人（j不超过30）

(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人（j不超过30）;

(3)Query i j ,i和j为正整数,i<=j，表示询问第i到第j个营地的总人数;

(4)End 表示结束，这条命令在每组数据最后出现;

 

#include <iostream>

#include <cstdio>

#include <cstring>

#define Max 50005

#define LL(x) ((x)<<1)

#define RR(x) ((x)<<1|1)

#define FF(i,n) for(int i = 0 ; i < n ; i ++)

using namespace std;

struct Node

{

int left,right,num;

int calmid()

{

return (left+right)/2;

}

}tt[Max<<2];

int num[Max];             //当left==right时,元线段。叶子节点

int build(int left,int right,int idx)

{

tt[idx].left = left;

tt[idx].right = right;

if(left == right)

{

   //printf("left = right = %d   :",left);

   //printf("tt[%d].num = %d\n",idx,num[left]);

return tt[idx].num = num[left];

}

int mid = (left + right)/2;

return tt[idx].num = build(left,mid,LL(idx)) + build(mid+1,right,RR(idx));

}

void update(int id,int x,int idx)               //从 idx 号节点往下更新，第id是元线段

{

tt[idx].num += x;                           //num + = x (每次加 X)

if(tt[idx].left == tt[idx].right)           //递归进行，直到查找到元线段

{

return ;

}

int mid = tt[idx].calmid();

if(id <= mid)

{

update(id,x,LL(idx));       //      /2

}

else

    {

update(id,x,RR(idx));       //      /2+1

}

}

int query(int left,int right,int idx)

{

if(left == tt[idx].left && right == tt[idx].right)

{

return tt[idx].num;

}

int mid = tt[idx].calmid();

if(right <= mid)

{

return query(left,right,LL(idx));

}

else if(mid < left)

{

return query(left,right,RR(idx));

}

else

{

return query(left,mid,LL(idx)) + query(mid+1,right,RR(idx));

}

}

int main()

{

    //freopen("in.txt","r",stdin);

    int T;

    scanf("%d",&T);

    for(int i=0;i<T;i++)

    {

        int n;

        scanf("%d",&n);

        for(int j=1;j<=n;j++)

        {

            scanf("%d",&num[j]);

            //printf("num[%d] = %d\n",j,num[j]);

        }

        build(1,n,1);

        printf("Case %d:\n",i+1);

       char com[9];

       while(scanf("%s",com))

       {

            if(strcmp(com,"End") == 0) break;

            int a,b;

            scanf("%d%d",&a,&b);

            switch(com[0])

            {

                   case 'Q':

                        printf("%d\n",query(a,b,1));

                        break;

                   case 'A':

update(a,b,1);

break;

   case 'S':

update(a,-b,1);

break;

     }

}

    }

    return 0;

}