POJ2828 Buy Tickets (树状数组 or 线段树) 插队问题

题目链接: Buy Tickets

大致题意

有N个人排队，每一个人都有一个val来对应，每一个后来人都会插入当前队伍的某一个位置pos。要求把队伍最后的状态输出。

解题思路

这里放一道类似的题目. Lost Cows

二分+树状数组 or 线段树

对于这种插队问题, 我们都需要采用逆向思维, 从后往前去想. 这样可以保证当前人所在的位置不会被之后的人挤掉.

举个例子: 如果有10个人, 第二个人说他要排在第五个位置, 假设这时候第三个人说他要排在第二个位置, 那你会发现, 如果你从前往后去想, 第二个人的位置是不是此时已经由5变到6了?
但是如果我们从后往前去看, 就不会有这样的问题.

具体做法就是, 我们遍历到第i个人的时候, 假设他说要排在第k个位置后, 则你就需要找到当前序列在哪个位置, 前面刚好有k个空位, 这时候我们就找到了这个人应该在的位置. (因为我们从后往前枚举, 如果前面有些位置有人了, 说明是这些人把当前这个人往后挤了, 不算是k个位置之内的人.)

我们可以用 线段树, 或者 树状数组 来维护.

AC代码

/* 线段树 */
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 2E5 + 10;
int a[N], b[N], res[N];

struct node {
	int l, r;
	int val; //区间有多少个空的
}t[N << 2];
void pushup(int x) { t[x].val = t[x << 1].val + t[x << 1 | 1].val; }

void build(int l, int r, int x = 1) {
	t[x] = { l, r, 1 };
	if (l == r) return;
	int mid = l + r >> 1;
	build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
	pushup(x);
}

void modify(int a, int c, int x = 1) {
	if (t[x].l == t[x].r) {
		t[x].val = c;
		return;
	}
	int mid = t[x].l + t[x].r >> 1;
	modify(a, c, x << 1 | (a > mid));
	pushup(x);
}

int ask(int k, int x = 1) {
	if (t[x].l == t[x].r) return t[x].l;
	if (t[x << 1].val >= k) return ask(k, x << 1);
	return ask(k - t[x << 1].val, x << 1 | 1);
}
int main()
{
	int n;
	while (~scanf("%d", &n)) {
		rep(i, n) scanf("%d %d", &a[i], &b[i]);
		build(1, n);
		for (int i = n; i >= 1; --i) {
			int pos = ask(a[i] + 1);
			modify(pos, 0);
			res[pos] = b[i];
		}

		rep(i, n) printf("%d%c", res[i], " \n"[i == n]);
	}

	return 0;
}



/* 树状数组 + 二分 */
int a[N], b[N], res[N];
int t[N];
int lowbit(int x) { return x & -x; }
void add(int x, int c) { for (int i = x; i < N; i += lowbit(i)) t[i] += c; }
int ask(int x) {
	int res = 0;
	for (int i = x; i; i -= lowbit(i)) res += t[i];
	return res;
}
int main()
{
	int n;
	while (~scanf("%d", &n)) {
		rep(i, n) {
			t[i] = lowbit(i);
			scanf("%d %d", &a[i], &b[i]);
		}

		for (int i = n; i >= 1; --i) {
			int l = 1, r = n;
			while (l < r) {
				int mid = l + r >> 1;
				if (ask(mid) >= a[i] + 1) r = mid;
				else l = mid + 1;
			}
			res[l] = b[i];
			add(l, -1);
		}

		rep(i, n) printf("%d%c", res[i], " \n"[i == n]);
	}
	return 0;
}

END