FZU 2171 —— 防守阵地 II（线段树，区间求和+区间更新）

题目：http://acm.fzu.edu.cn/problem.php?pid=2171

给定N个数和一个M，Q个查询，每个查询都只有一个X，对于每个查询，先输出[X, X+M-1]这个区间所有数的和，然后将这个区间每个数字都减一。

典型的区间求和，区间更新，果断拍个线段树给它！

#include<cstdio>
#include<cstring>
#define N 100001
#define lson o<<1
#define rson (o<<1)|1
int a[N];
int s[N<<2], l[N<<2], r[N<<2], f[N<<2];
void update(int o, int ll,int rr, int v);
void pushdown(int o){
    if(f[o]){
        int m = (l[o]+r[o])>>1;
        update(lson, l[o], m, f[o]);
        update(rson, m+1, r[o], f[o]);
        f[o]=0;
    }
}
void maintain(int o){
    s[o] = s[lson]+s[rson];
}
void build(int o,int ll, int rr){
    l[o]=ll; r[o]=rr;
    s[o]=f[o]=0;
    if(ll<rr){
        int m=(ll+rr)>>1;
        build(lson, ll, m);
        build(rson, m+1, rr);
        maintain(o);
    }
    else{
        s[o] = a[ll];
    }
}
void update(int o,int ll,int rr, int v){
    if(l[o]==ll && r[o]==rr){
        int len = rr-ll+1;
        s[o]+=v*len;
        f[o]+=v;
        return;
    }
    pushdown(o);
    int m = (l[o]+r[o])>>1;
    if(rr<=m)   update(lson, ll, rr, v);
    else if(ll>m)   update(rson, ll, rr, v);
    else{
        update(lson, ll, m, v);
        update(rson, m+1, rr, v);
    }
    maintain(o);
}
int query(int o, int ll, int rr){
    if(l[o]==ll && r[o]==rr)    return s[o];
    pushdown(o);
    int m = (l[o]+r[o])>>1;
    int tmp;
    if(rr<=m)   tmp = query(lson, ll, rr);
    else if(ll>m)   tmp = query(rson, ll, rr);
    else    tmp = query(lson, ll, m) + query(rson, m+1, rr);
    maintain(o);
    return tmp;
}
int main(){
    int n, m, q;
    int x;
    while(~scanf("%d %d %d", &n, &m, &q)){
        for(int i=1; i<=n; i++) scanf("%d",a+i);
        build(1, 1, n);
        while(q--){
            scanf("%d", &x);
            printf("%d\n", query(1, x, x+m-1));
            update(1, x, x+m-1, -1);
        }
    }
    return 0;
}