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
部队中总共有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<string.h>
#include<stdio.h>
using namespace std;
#define Max 100010
int a[Max],m,n,q;
struct node
{
int l,r,sum,data;
}tree[4*Max];
void build(int l,int r,int nd)
{
tree[nd].l=l,tree[nd].r=r,tree[nd].sum=a[l],tree[nd].data=0;
if(l==r)
{
return;
}
int mid=(l+r)/2;
build(l,mid,2*nd+1);
build(mid+1,r,2*nd+2);
tree[nd].sum=tree[2*nd+1].sum+tree[2*nd+2].sum;
}
void update(int l,int r,int nd)
{
if(tree[nd].l==l&&tree[nd].r==r)
{
tree[nd].sum-=(r-l+1);
tree[nd].data++;
return;
}
int mid=(tree[nd].l+tree[nd].r)/2;
if(l<=mid)
{
if(r>=mid+1)
{
update(l,mid,2*nd+1);
update(mid+1,r,2*nd+2);
}
else update(l,r,2*nd+1);
}
else update(l,r,2*nd+2);
tree[nd].sum=tree[2*nd+1].sum+tree[2*nd+2].sum;
}
int query(int l,int r,int nd)
{
if(tree[nd].l==l&&tree[nd].r==r)
return tree[nd].sum;
int mid=(tree[nd].l+tree[nd].r)/2;
if(tree[nd].data)
{
tree[2*nd+1].data+=tree[nd].data;
tree[2*nd+1].sum-=tree[nd].data*(tree[2*nd+1].r-tree[2*nd+1].l+1);
tree[2*nd+2].data+=tree[nd].data;
tree[2*nd+2].sum-=tree[nd].data*(tree[2*nd+2].r-tree[2*nd+2].l+1);
tree[nd].data=0;
}
if(l<=mid)
{
if(r>=mid+1)
return query(l,mid,2*nd+1)+query(mid+1,r,2*nd+2);
else return query(l,r,2*nd+1);
}
else return query(l,r,2*nd+2);
}
int main()
{
//freopen("b.txt","r",stdin);
int b;
while(scanf("%d %d %d",&n,&m,&q)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,0);
while(q--)
{
scanf("%d",&b);
printf("%d\n",query(b,b+m-1,0));
update(b,b+m-1,0);
}
}
return 0;
}