HDU 1166 敌兵布阵

题目：敌兵布阵

思路：
线段树裸题~
区间减可以看做区间加一个相反数。

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 50000

int n,m;
int a[maxn*10+5];
int c[maxn+5];
int P,Q;

void Add(int o,int L,int R,int k) {
    if(P==L&&L==R) {
        a[o]+=k;
        return ;
    }
    if(P>R||P<L) return ;
    int mid=L+(R-L)/2;
    int lch=o*2,rch=o*2+1;
    Add(lch,L,mid,k),Add(rch,mid+1,R,k);
    a[o]=a[lch]+a[rch];
    return ;
}

int Query(int o,int L,int R) {
    if(P<=L&&R<=Q) {
        return a[o];
    }
    if(P>R||Q<L) return 0;
    int mid=L+(R-L)/2;
    int lch=o*2,rch=o*2+1;
    return Query(lch,L,mid)+Query(rch,mid+1,R);
}

void make_tree(int o,int L,int R) {
    if(L==R) {
        a[o]=c[L];
        return ;
    }
    int mid=L+(R-L)/2;
    int lch=o*2,rch=o*2+1;
    make_tree(lch,L,mid),make_tree(rch,mid+1,R);
    a[o]=a[lch]+a[rch];
}

int main() {
    int T;
    scanf("%d",&T);
    for(int t=1; t<=T; t++) {
        memset(a,0,sizeof(a));
        printf("Case %d:\n",t);
        scanf("%d",&n);
        for(int i=1; i<=n; i++) {
            scanf("%d",&c[i]);
        }
        make_tree(1,1,n);
        while(true) {
            char opr[10];
            scanf("%s",opr);
            if(opr[0]=='E') break;
            scanf("%d%d",&P,&Q);
            if(opr[0]=='Q') {
                int ans=Query(1,1,n);
                printf("%d\n",ans);
            } else {
                int k=Q;
                if(opr[0]=='S') k=-k;
                Add(1,1,n,k);
            }
        }
    }
    return 0;
}

数据生成器：

#include<bits/stdc++.h>
using namespace std;

#define maxn 50000
#define maxm 50000
#define maxa 50
#define maxT 10
#define Rand() (rand()+rand()%1000000007)

int main() {
    srand(time(NULL));
    int T;
    T=Rand()%maxT+1;
    printf("%d\n",T);
    while(T--) {
        int n,m;
        n=Rand()%maxn+1;
        m=Rand()%maxm+1;
        printf("%d\n",n);
        for(int i=1;i<=n;i++) printf("%d ",Rand()%maxn+1);
        printf("\n");
        while(m--) {
            int opr=Rand()%3;
            if(opr==0) printf("Query ");
            else if(opr==2)printf("Add ");
            else printf("Sub ");
            int l=1,r=0;
            while(l>r) {
                l=Rand()%n+1;
                r=n-Rand()%n;
            }
            printf("%d %d\n",l,r);
        }
        printf("End\n");
    }

    return 0;
}