防守阵地 II

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

输出：

6

3

5

//线段树查找区间，更新区间

线段树+延迟更新

#include <stdio.h>
#define M 100005
int a[M];
struct data
{
    int l,r;
    int add,val,cover;
}node[3*M];

void BuildTree(int left,int right,int u)
{
    node[u].l=left;
    node[u].r=right;
    node[u].add=0;
    node[u].cover=0;
    node[u].val=0;
	int mid=(left+right)>>1,i;
	for(i=left;i<=right;i++) node[u].val+=a[i];
	if(right-left>=1)  
	{
     	BuildTree(left,mid,u<<1);    
    	BuildTree(mid+1,right,(u<<1)+1);   
	}
}
void updata(int left,int right,int u)//更新 
{
    if (node[u].l==left&&node[u].r==right)  //成段更新
    {
        node[u].add++;
        node[u].cover=1;
        node[u].val-=(node[u].r - node[u].l+1);
      return ;
    }
     int mid = (node[u].l + node[u].r)>>1;
       if(right <= mid)
           updata (left,right,u<<1);
       else if(left >= mid+1)
          updata(left,right,(u<<1)+1);
      else
       {
        updata(left,mid,u<<1);
        updata(mid+1,right,(u<<1)+1);
       }
    node[u].val = node[u<<1].val + node[(u<<1)+1].val;    //递归回来 修改父亲结点的 val
}
void get_down (int u)        //延迟覆盖
{
    node[u<<1].add+=node[u].add;
    node[u<<1].val-=(node[u<<1].r-node[u<<1].l+1)*node[u].add;
    node[u<<1].cover=1;
    
    node[(u<<1)+1].add+=node[u].add;
    node[(u<<1)+1].val-=(node[(u<<1)+1].r-node[(u<<1)+1].l+1)*node[u].add;
    node[(u<<1)+1].cover = 1;
    node[u].add = 0;
    node[u].cover=0; 
}
int Query(int left,int right,int u)//查找 
{
     if(node[u].l==left&&node[u].r==right) 
     return node[u].val;
   
     if(node[u].cover)   //延迟更新 
     { get_down(u);  
    } 
     int mid=(node[u].l+node[u].r)>>1,ret;
	if(left<=mid)
	{
		if(right>=mid+1)
			return Query(left,mid,u<<1)+Query(mid+1,right,(u<<1)+1);
		else return Query(left,right,u<<1);
	}
	else return Query(left,right,(u<<1)+1);
}
int main()
{
	int x,i,n,m,q;
	while(scanf("%d%d%d",&n,&m,&q)!=EOF)
	{
		for(i=1;i<=n;i++)
			scanf("%d",&a[i]);
		BuildTree(1,n,1);
		for(i=0;i<q;i++)
		{
			scanf("%d",&x);
		     if(x>=1&&x<=n-m+1)
        	{
             printf("%d\n",Query(x,x+m-1,1));
			 updata(x,x+m-1,1);
		   }
        }
	}
	return 0;
}