POJ 2828 线段树单点更新

POJ 2828

题解：这题最巧的是， 最后一个人的顺序是固定的， 反向更新运用线段树比较简单。

比如， 最后一名选在第i个人后面， 那么他的位置就是i + 1, 以此类推倒数第二个.....

这题是早前训练做的一道， 今天补上题解

AC code：

//adruill's submission
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<iostream>
#include<bitset>
#include<map>
using namespace std;

#define M(a, b) memset(a, b, sizeof(a))
#define mid ((l + r) >> 1)
#define ls rt << 1, l, mid
#define rs rt << 1|1, mid + 1, r
#define lowbit(x) (x & (-x))
#define REP(n) for(int i = 1; i <= n; i++)
#define debug 0

const int maxn = 200000 + 5;

int n, sum[maxn << 2], ans[maxn << 2], pos[maxn], v[maxn];			//注意ans[maxn << 2]这个空间

void pushUp(int rt) {
	sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];						//子结点信息更新到父节点
}

void build(int rt, int l, int r) {
	sum[rt] = r - l + 1;											//建树， sun[rt] 保存[l, r]未被占用量
	if (l == r)	
		return;
	
	build(ls);
	build(rs);
	pushUp(rt);
}

void update(int p, int v, int rt, int l, int r) {
	if (l == r) {											//更新， 找到位置， 叶子节点sum[rt]更新， 赋值
		sum[rt]--;
		ans[rt] = v;
		return;
	}
		
	if (p <= sum[rt << 1])									//因为p代表要放在当前[l ,r]区间内第p个空位置
		update(p, v, ls);									//如果左孩子有空位，找左，反之找有孩子
	else
		update(p - sum[rt << 1], v, rs);					//注意这里 p - sum[rt << 1]， 做孩子的空位置不能忽略

	pushUp(rt);
}

void print(int rt, int l, int r) {
	if (l == r) {
		printf("%s%d", l == 1 ? "" : " ",  ans[rt]);		//输出
		return;
	}

	print(ls);
	print(rs);
}
int main() {
#if debug
	freopen("in.txt", "r", stdin);
	//freopen("out.txt", "w", stdout);
#endif //debug
	
	while (~scanf("%d", &n)) {
		M(sum, 0);		
		build(1, 1, n);										//建树
		for (int i = 1; i <= n; i++)
			scanf("%d%d", &pos[i], &v[i]);
		for (int i = n; i >= 1; i--)						//反向更新
			update(pos[i] + 1, v[i], 1, 1, n);

		print(1, 1, n);
		puts("");								
	}
	return 0;
}