POJ 1442 Black Box

题目链接

参考资料：陈启峰《Size Balanced Tree》

题目大意：给一个整数序列A(n)(n≤30000)，一个不降的正整数序列u(m)(m≤30000)，依次求A(n)中前u(i)个数的第i大的值。题目实际读取数据的方式和上面的陈述略有不同。

分析：简单来说，就是要执行n次插入操作和m次查询操作。可以用树状数组，离散化之后每次二分查找求第i大的数。更容易想到的是二叉树，但是普通二叉树会超时，因为可能遇到二叉树退化为链的情况。突然想到很久之前(好几年前了)看过的SBT，于是翻出来又看一遍，然后AC了。

SBT的代码还是有点复杂，更好写的算法是Treap。就当是练习了。

const int MAX_N = 30000 + 5;
int ip = 0;
int key[MAX_N];
int s[MAX_N];
int lft[MAX_N];
int rht[MAX_N];
int a[MAX_N];

void right_rotate(int &t)
{
	int k = lft[t];
	lft[t] = rht[k];
	rht[k] = t;
	s[k] = s[t];
	s[t] = s[lft[t]] + s[rht[t]] + 1;
	t = k;
}

void left_rotate(int &t)
{
	int k = rht[t];
	rht[t] = lft[k];
	lft[k] = t;
	s[k] = s[t];
	s[t] = s[lft[t]] + s[rht[t]] + 1;
	t = k;
}

void maintain(int &t, bool flag)
{
	if (flag)
	{
		if (s[rht[rht[t]]] > s[lft[t]]) left_rotate(t);
		else if (s[lft[rht[t]]] > s[lft[t]])
		{
			right_rotate(rht[t]);
			left_rotate(t);
		}
		else return;
	}
	else
	{
		if (s[lft[lft[t]]] > s[rht[t]]) right_rotate(t);
		else if (s[rht[lft[t]]] > s[rht[t]])
		{
			left_rotate(lft[t]);
			right_rotate(t);
		}
		else return;
	}
	maintain(lft[t], false);
	maintain(rht[t], true);
	maintain(t, false);
	maintain(t, true);
}

int new_node(int v)
{
	key[++ip] = v;
	s[ip] = 1;
	return ip;
}

void sbt_insert(int &t, int v)
{
	if (t == 0) t = new_node(v);
	else
	{
		int par = 0, cur = t;
		while (cur)
		{
			++s[par = cur];
			if (v <= key[cur]) cur = lft[cur];
			else cur = rht[cur];
		}
		if (v <= key[par]) lft[par] = new_node(v);
		else rht[par] = new_node(v);
		maintain(t, v > key[t]);
	}
}

int find_kth(int t, int k)
{
	if (k == s[lft[t]] + 1) return key[t];
	else if (k <= s[lft[t]]) return find_kth(lft[t], k);
	else return find_kth(rht[t], k - s[lft[t]] - 1);
}

int main(int argc, char *argv[])
{
	int n, m, p;
	cin >> n >> m;
	for (int i = 0; i < n; ++i) scanf("%d", a + i);
	int pos = 0;
	int root = 0;
	for (int i = 1; i <= m; ++i)
	{
		scanf("%d", &p);
		while (pos < p) sbt_insert(root, a[pos++]);
		printf("%d\n", find_kth(root, i));
	}
	return 0;
}