FZU Problem 2171 防守阵地 II (线段树，区间更新)

时间：2014-08-10 01:42:09 阅读：289 评论：0 收藏：0 [点我收藏+]

标签：des style blog http color os io strong

Problem 2171 防守阵地 II

Accept: 143 Submit: 565
Time Limit: 3000 mSec Memory Limit : 32768 KB

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

Sample Output

线段树区间修改，注意延迟标记，修改函数和查询函数的参数是不能改的啊！

延迟标记：

每个节点新增加一个标记，记录这个节点是否进行了某种修改(这种修改操作会影响其子节点)，对于任意区间的修改，我们先按照区间查询的方式将其划分成线段树中的节点，然后修改这些节点的信息，并给这些节点标记上代表这种修改操作的标记。在修改和查询的时候，如果我们到了一个节点p，并且决定考虑其子节点，那么我们就要看节点p是否被标记，如果有，就要按照标记修改其子节点的信息，并且给子节点都标上相同的标记，同时消掉节点p的标记。

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<stdlib.h>
  4 #include<algorithm>
  5 using namespace std;
  6 const int MAXN=100000+10;
  7 int b[MAXN*4];
  8 struct node
  9 {
 10     int l,r;
 11     int num;
 12     int col;
 13     int mid()
 14     {
 15         return (l+r)/2;
 16     }
 17 }a[MAXN*5];
 18 
 19 void pushup(int step)
 20 {
 21     a[step].num=a[step*2].num+a[step*2+1].num;
 22 }
 23 
 24 void pushdown(int x,int step)
 25 {
 26     if(a[step].col!=0)
 27     {
 28         a[step*2+1].col+=a[step].col;
 29         a[step*2].col+=a[step].col;
 30         a[step*2].num+=a[step].col*(x-(x/2));
 31         a[step*2+1].num+=a[step].col*((x/2));
 32         a[step].col=0;
 33     }
 34 }
 35 
 36 void btree(int l,int r,int step)
 37 {
 38     a[step].l=l;
 39     a[step].r=r;
 40     a[step].col=0;
 41     if(l==r)
 42     {
 43         a[step].num=b[l];
 44         return ;
 45     }
 46     int mid=a[step].mid();
 47     btree(l,mid,step*2);
 48     btree(mid+1,r,step*2+1);
 49     pushup(step);
 50 }
 51 
 52 void ptree(int l,int r,int val,int step)
 53 {
 54     if(l<=a[step].l&&a[step].r<=r)
 55     {
 56         a[step].col+=val;
 57         a[step].num+=val*(a[step].r-a[step].l+1);
 58         return ;
 59     }
 60     pushdown(a[step].r-a[step].l+1,step);
 61     int mid=a[step].mid();
 62     if(l>mid)
 63         ptree(l,r,val,step*2+1);
 64     else if(r<=mid)
 65         ptree(l,r,val,step*2);
 66     else
 67     {
 68         ptree(l,r,val,step*2);
 69         ptree(l,r,val,step*2+1);
 70     }
 71     pushup(step);
 72 }
 73 
 74 int fintree(int l,int r,int step)
 75 {
 76     if(l<=a[step].l&&a[step].r<=r)
 77          return a[step].num;
 78     pushdown(a[step].r-a[step].l+1,step);
 79     int mid=a[step].mid();
 80     if(l>mid)
 81         return fintree(l,r,step*2+1);
 82     else if(r<=mid)
 83         return fintree(l,r,step*2);
 84     else
 85         return fintree(l,r,step*2)+fintree(l,r,step*2+1);
 86 }
 87 
 88 int main()
 89 {
 90     int n,m,kase,num;
 91     while(scanf("%d %d %d",&n,&m,&kase)!=EOF)
 92     {
 93         for(int i=1;i<=n;i++)
 94             scanf("%d",&b[i]);
 95         btree(1,n,1);
 96         while(kase--)
 97         {
 98             scanf("%d",&num);
 99             int ans=fintree(num,num+m-1,1);
100             printf("%d\n",ans);
101             ptree(num,num+m-1,-1,1);
102         }
103     }
104     return 0;
105 }