线段树 HDU1166 敌兵布阵

题目链接：http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=16216

题意：给定每个点的人数，并且中间会有增减的操作。求某一段区间的人数和

思路：线段树第一题，所以多少有点参考别人代码。主要熟悉了下线段树的建立、查找、更新的方法，采用二进制。

源码：

#include <cstdio>

#include <cstring>

#include <cmath>

#include <algorithm>

int const MAXN = 50000+5;

int peo[MAXN<<2];

int n;

int query(int l,int r,int L,int R,int st)

{

    int ans = 0;

    if(l<=L && r>=R)

        return peo[st];

    int mid = (L+R)>>1;

    if(mid>=l)  ans+=query(l,r,L,mid,st<<1);

    if(mid<r)   ans+=query(l,r,mid+1,R,(st<<1)+1);

//    printf("ans = %d\n",ans);

    return ans;

}

void update(int mark,int add,int L,int R,int st)

{

    if(L==R){

        peo[st] += add;

        return;

    }

    else{

        int mid = (R+L)/2;

        int ans = 0;

        if(mid>=mark) update(mark,add,L,mid,st<<1);

        else update(mark,add,mid+1,R,(st<<1)+1);

//        printf("ans = %d\n",ans);

        peo[st] = peo[st<<1] + peo[(st<<1)+1];

        return;

    }

//    for(int i=1;i<=(n<<2); i++)

//        printf("%d ",peo[i]);

//    printf("\n");

}

int main()

{

    int t;

    scanf("%d",&t);

    for(int i=1; i<=t; i++){

        memset(peo,0,sizeof(peo));

        int a,b;

        char temp[5];

        scanf("%d",&n);

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

            scanf("%d",&b);

            update(j,b,1,n,1);

        }

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

        while(scanf("%s",temp)!=EOF && temp[0]!='E'){

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

            if(temp[0]=='Q')

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

            if(temp[0] == 'A') update(a,b,1,n,1);

            else if(temp[0] == 'S') update(a,-b,1,n,1);

//            for(i=1; i<(n<<2); i++)

//                printf("%d ",peo[i]);

//            printf("\n");

        }

    }

    return 0;

}