【POJ2182】Lost Cows

【POJ2182】Lost Cows

题面

vjudge

题解

从后往前做

每扫到一个点\(i\)以及比前面小的有\(a[i]\)个数

就是查询当前的第\(a[i]+1\)小

然后查询完将这个数删掉

两个操作可以用平衡树实现

但是我比较懒用了\(01trie\)

据说暴力也可以过

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
inline int gi() {
    register int data = 0, w = 1;
    register char ch = 0;
    while (ch != '-' && (ch > '9' || ch < '0')) ch = getchar();
    if (ch == '-') w = -1 , ch = getchar();
    while (ch >= '0' && ch <= '9') data = data * 10 + (ch ^ 48), ch = getchar();
    return w * data;
}
const int MAX_N = 80005;
struct Trie { int ch[2], size; } t[MAX_N * 18]; int tot;
void insert(int v) {
	int o = 1; t[o].size++;
	for (int i = 17; i >= 0; i--) {
		int c = v >> i & 1;
		if (!t[o].ch[c]) t[o].ch[c] = ++tot;
		t[o = t[o].ch[c]].size++;
	}
}
void erase(int v) {
	int o = 1; t[o].size--;
	for (int i = 17; i >= 0; i--) {
		int c = v >> i & 1;
		t[o = t[o].ch[c]].size--;
	}
}
int Kth(int k) {
	int o = 1, res = 0;
	for (int i = 17; i >= 0; i--) {
		int sz = t[t[o].ch[0]].size;
		if (k <= sz) o = t[o].ch[0];
		else k -= sz, res |= 1 << i, o = t[o].ch[1];
	}
	return res;
}
int N, a[MAX_N], ans[MAX_N];
int main () {
	N = gi(); for (int i = 1; i < N; i++) a[i + 1] = gi(); ++tot;
	for (int i = 1; i <= N; i++) insert(i);
	for (int i = N; i >= 1; i--) {
		ans[i] = Kth(a[i] + 1);
		erase(ans[i]);
	}
	for (int i = 1; i <= N; i++) printf("%d\n", ans[i]);
    return 0;
}