防守阵地 II【线段树 区间修改】

E - 防守阵地 II
Time Limit:3000MS     Memory Limit:32768KB     64bit IO Format:%I64d
& %I64u
Submit Status Practice FZU
2171

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 32 1 3 1 4123

Sample Output

635

FAQ | About Virtual Judge | Forum | Discuss | Open
Source Pro

模板

#include<stdio.h>
#define lson l , m ,rt << 1
#define rson m+1 , r , rt << 1 | 1
const int maxn = 100002;
int add[maxn << 2];
int 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("%d",&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] += 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);
}

int 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;
int ret = 0;
if(L <= m) ret+=query(L , R , lson);
if(m < R) ret+=query(L , R , rson);
return ret;
}

int main()
{
int n,m,q;
while(~scanf("%d%d%d",&n,&m,&q))
{
build(1,n,1);
int ans,x;
while(q--)
{
scanf("%d",&x);
ans=query(x,x+m-1,1,n,1);
update(x,x+m-1,-1,1,n,1);
printf("%d\n",ans);
}
}
return 0;
}