FZU 2171 防守阵地 II（线段树区间更新【lazy标记】）

Problem Description

部队中总共有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

本题是一道线段树区间更新的裸题，裸题的意思呢就是直接套模板就能AC的题，在这里用到了lazy标记。lazy标记的用处就是减少区间更新所花费分时间，提高代码的效率。
代码中有详细的介绍。
下面是AC代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

struct node
{
    int left,right,val,lazy,len;
} c[300005];

void build_tree(int root,int l,int r)
{
    c[root].left=l;
    c[root].right=r;
    c[root].lazy=0;//用来标记区间更新的次数
    c[root].len=c[root].right-c[root].left+1;//计算出每个区间的长度，方便以后的区间更新
    if(c[root].left==c[root].right)
    {
        scanf("%d",&c[root].val);
        return ;
    }
    int mid=(c[root].left+c[root].right)/2;
    build_tree(root*2,l,mid);
    build_tree(root*2+1,mid+1,r);
    c[root].val=c[root*2].val+c[root*2+1].val;//表示每个区间的价值是其两个子区间的价值之和
}

void pushdown(int root)
{
    c[root*2].lazy+=c[root].lazy;//子节点需要更新的次数等于父节点需要更新的次数，下同
    c[root*2+1].lazy+=c[root].lazy;
    c[root].lazy=0;
    if(c[root*2].left==c[root*2].right&&c[root*2].lazy>0)
    {
        c[root*2].val-=c[root*2].lazy;//给子节点进行更新
        c[root*2].lazy=0;
    }
    if(c[root*2+1].left==c[root*2+1].right&&c[root*2+1].lazy>0)
    {
        c[root*2+1].val-=c[root*2+1].lazy;
        c[root*2+1].lazy=0;
    }
}


int search_tree(int root,int l,int r)
{
    if(c[root].left==l&&c[root].right==r)
    {
        return c[root].val-c[root].len*c[root].lazy;
    }
    if(c[root].lazy!=0)//如果该节点的lazy=0则表示这个区间不需要更新
    {
        pushdown(root);
    }
    int cnt;
    int mid=(c[root].right+c[root].left)/2;
    if(mid<l)//用二分的方法计算某个区间的区间和
    {
        cnt=search_tree(root*2+1,l,r);
    }
    else if(mid>=r)
    {
        cnt=search_tree(root*2,l,r);
    }
    else
    {
        cnt=search_tree(root*2,l,mid)+search_tree(root*2+1,mid+1,r);
    }
    return cnt;
}

void update_tree(int root,int l,int r)//更新lazy
{
    if(c[root].left==l&&c[root].right==r)
    {
        c[root].lazy++;
        return ;
    }
    int mid=(c[root].left+c[root].right)/2;
    if(mid<l)
    {
        update_tree(root*2+1,l,r);
    }
    else if(mid>=r)
    {
        update_tree(root*2,l,r);
    }
    else
    {
        update_tree(root*2,l,mid);
        update_tree(root*2+1,mid+1,r);
    }
    c[root].val=c[root*2].val-c[root*2].lazy*c[root*2].len+c[root*2+1].val-c[root*2+1].lazy*c[root*2+1].len;
}

int main()
{
       int n,m,q;
       while(~scanf("%d%d%d",&n,&m,&q))
       {
           build_tree(1,1,n);
           while(q--)
           {
               int num;
               scanf("%d",&num);
               int re=search_tree(1,num,num+m-1);
               update_tree(1,num,num+m-1);
               printf("%d\n",re);
           }
       }
       return 0;
}