HDU1166 敌兵布阵 简单线段树 或树状数组

HDU1166 敌兵布阵

 

最近线段树做的多了，简单线段树。树状数组就比较容易了。冲冲冲！！！

下面我用树状数组写的。

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<algorithm>
#include<string>
#include<vector>
using namespace std;
int c[50010];
int n;
int lowbit(int x) {return x&(-x);}
void add(int x,int num)
{
    for(int i=x;i<=n;i+=lowbit(i))
        c[i]+=num;
}
int sum(int x)
{
    int ans=0;
    for(int i=x;i>0;i-=lowbit(i))
        ans+=c[i];
    return ans;
}
int main()
{
//    freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    for(int t=1;t<=T;t++)
    {
        memset(c,0,sizeof(c));
        printf("Case %d:\n",t);
        char op[10];
        int x,y;
        scanf("%d",&n);
        int num;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&num);
            add(i,num);
        }

        while(1)
        {
            scanf("%s",op);
            if(op[0]=='E')
                break;
            scanf("%d%d",&x,&y);
            if(op[0]=='A')
                add(x,y);
            if(op[0]=='S')
                add(x,-y);
            if(op[0]=='Q')
                printf("%d\n",sum(y)-sum(x-1));
        }

    }
    return 0;
}