FZU Problem 2171 防守阵地 II （线段树模板题）

Problem 2171 防守阵地 II

 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

代码：

/**
*  解题思路：线段树模板题
*/
#include <cstdio>
#define lson l , m , rt << 1
#define rson m + 1 , r , rt << 1 | 1
#define LL long long
const int maxn = 100005;
LL add[maxn << 2];
LL 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("%lld", &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] += (LL)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);
}
LL 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;
	LL 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) != EOF)
	{
		build(1, N, 1);
		int ans;
		for (int i = 1; i <= Q; i++)
		{
			int x;
			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;
}