HDUOJ 1166 敌兵布阵

单点更新+区间求和

#include<cstdlib>
#include<cstring>
#include<cstdio>
using namespace std;
const long M=50005;
struct Seb_tree
{
    long l;
    long r;
    long long sum;
};
Seb_tree t[M*4+1];
long s[M];
long long total=0;
long n,m;
void query(long x, long y, long k)
{
    if (t[k].l==x && t[k].r==y) total+=t[k].sum;
    else 
    {
        long mid=(t[k].l+t[k].r)/2;
        if (y<=mid) query(x, y, k*2);
        else if (x > mid) query(x, y, k*2+1);
        else 
        {
            query(x, mid, k*2);
            query(mid+1, y, k*2+1);
        }
    }
}
void sub(long x, long y, long k)
{
    t[k].sum-=y;
    if (t[k].l==x && t[k].r==x) return;
    else 
    {
        long mid=(t[k].l+t[k].r)/2;
        if ( x<=mid ) sub(x, y, k*2);
        else sub(x, y, k*2+1);
    }
}
void add(long x, long y, long k)
{
    t[k].sum+=y;
    if (t[k].l==x && t[k].r==x) return;
    else 
    {
        long mid=(t[k].l+t[k].r)/2;
        if (x<=mid) add(x, y, k*2);
        else add(x, y, k*2+1);
    }
}
void build(long l, long r, long k)
{
    t[k].l=l;
    t[k].r=r;
    if (t[k].l==t[k].r) t[k].sum=s[l];
    else 
    {
        long mid=(l+r)/2;
        build(l, mid, k*2);
        build(mid+1, r, k*2+1);
        t[k].sum=t[k*2].sum+t[k*2+1].sum;
    }
}
int main()
{
    long j=0;
    scanf("%d", &m);
    while (m--)
    {
    	memset(t, 0, sizeof(t));
    	memset(s, 0, sizeof(s));
        scanf("%d", &n);
        for (long i=1; i<=n; i++) scanf("%d", &s[i]);
        build(1, n, 1);
        printf( "Case %d:\n", ++j );
        char ss[8];
        while (scanf("%s", &ss))
        {
            if (ss[0]=='E') break;
            long t1, t2;
            if (ss[0]=='A') 
            {
    
                scanf("%d%d", &t1, &t2);
                add(t1, t2, 1);
            }
            else if (ss[0]=='S') 
            {
               scanf("%d%d", &t1, &t2);
               sub(t1, t2, 1);
            }
            else if (ss[0]=='Q')
            {
                total=0;
                scanf("%d%d", &t1, &t2);
                query(t1, t2, 1);
                printf("%I64d\n", total);
            }
        }
    }
    return 0;
}