fzu 2171 防守阵地 II(线段树 区间更新)

部队中总共有N个士兵，每个士兵有各自的能力指数Xi，在一次演练中，指挥部确定了M个需要防守的地点，指挥部将选择M个士兵依次进入指定地点进行防守任务，获得的参考指数即为M个士兵的能力之和。随着时间的推移，指挥部将下达Q个指令来替换M个进行防守的士兵们，每个参加完防守任务的士兵由于疲惫等原因能力指数将下降1。现在士兵们排成一排，请你计算出每次进行防守的士兵的参考指数。

Input

输入包含多组数据。

输入第一行有两个整数N,M,Q(1<=N<=100000，1<=M<=1000,1<=Q<=100000),第二行N个整数表示每个士兵对应的能力指数Xi（1<=Xi<=1000）。

接下来Q行，每行一个整数X，表示在原始队列中以X为起始的M个士兵替换之前的士兵进行防守。（1<=X<=N-M+1）

对于30%的数据1<=M,N,Q<=1000。

Output

输出Q行，每行一个整数，为每次指令执行之后进行防守的士兵参考指数。

Sample Input

5 3 3

2 1 3 1 4

1

2

3

Sample Output

6

3

5

思路：线段树的延迟更新，一开始查完就更新，超时

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int maxn=111111;
typedef long long  LL;
LL  add[maxn<<2];
LL sum[maxn<<2];
void PushUp(int rt){
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];//更新父节点的值
}

void PushDown(int rt,int m){//向下更新值
    if(add[rt]){
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];//代表更新的数值
        sum[rt<<1]+=add[rt]*(m-(m>>1));//当前区间段需要更新的
        sum[rt<<1|1]+=add[rt]*(m>>1);
        add[rt]=0;//往下更新过之后就不用标记向下更新
    }
}

void build (int l,int r,int rt){
    add[rt]=0;
    if(l==r){
        scanf("%lld",&sum[rt]);
        return ;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt);
}
void update(int L,int R,int c,int l,int r,int rt){
    if(L<=l && r<=R){
        add[rt]+=c;
        sum[rt]+=LL(c)*(r-l+1);//更新到区间节点的时候就不用再更新了
        return ;
    }
    PushDown(rt,r-l+1);//如果要更新的点在上一次更新的下面，要先把上次没更新的更新
    int m=(l+r)>>1;
    if(L<=m) update(L,R,c,lson);
    if(m<R) update(L,R,c,rson);
    PushUp(rt);
}
long long  query(int L,int R,int l,int r,int rt){
    if(L<=l && r<=R){
        return sum[rt];
    }
    PushDown(rt,r-l+1);。
    int m=(l+r)>>1;
    LL ret=0;
    if(L<=m) ret+=query(L,R,lson);
    if(R>m) ret+=query(L,R,rson);
    return ret;
}
 int main(){
     int n,m,q;
     LL x;
     while(scanf("%d%d%d",&n,&m,&q)!=EOF){
         build(1,n,1);
         while(q--){
            scanf("%lld",&x);
            printf("%lld\n",query(x,x+m-1,1,n,1));//查询
            update(x,x+m-1,-1,1,n,1);

         }
     }
     return 0;
 }