BZOJ3224 普通平衡树

题目传送门

Description

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

Input

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)

Output

对于操作3,4,5,6每行输出一个数，表示对应答案

题解

这道题很明显要用二叉排序树来进行操作，其实是裸题。不过以前没打过splay，考试时教练让我们打splay。没打出来，调了一晚上也没调出来。wwwww。熬夜打了一个treap，还要上课啊。。。。。补觉了。

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <ctime>
using namespace std;
const int Maxn = 101111;
struct node {
	int s,v,r,cnt; int ch[2];
}N[Maxn];
int totN, root;
const int INF = 1 << 30;
void up(int u) {
	int ls = N[u].ch[0], rs = N[u].ch[1];
	N[u].s = N[u].cnt + N[ls].s + N[rs].s;
}

void Rot(int &u, int d) {
	int t = N[u].ch[d^1];  N[u].ch[d^1] = N[t].ch[d];
	N[t].ch[d] = u;
	up(u); up(t); u = t;
}

void Ins(int& u,int val) {
	if (!u) {
		u = ++totN; N[u].v = val; N[u].cnt = 1; N[u].s = 1; N[u].r = rand();
	} else {
		++N[u].s; int k = val < N[u].v ? 0 : 1;
		if (N[u].v == val) { ++N[u].cnt; return; }
		Ins(N[u].ch[k], val);
		if (N[N[u].ch[k]].r > N[u].r) Rot(u, k ^ 1);
	}
}

void Del(int& u,int val) {
	if (!u) return;
	if (N[u].v == val) {
		if (N[u].cnt > 1) { --N[u].cnt; --N[u].s; return; }
		if (!N[u].ch[0] || !N[u].ch[1]) u = N[u].ch[0] + N[u].ch[1];
		else {
			int k = N[N[u].ch[0]].r > N[N[u].ch[1]].r;
			Rot(u,k); Del(u, val); 
		}
	} else --N[u].s, Del(N[u].ch[val < N[u].v ? 0 : 1],val);
}

int Rank(int u, int val) {
	if (!u) return 0;
	if (N[u].v == val) return N[N[u].ch[0]].s + 1;
	if (N[u].v < val) return N[N[u].ch[0]].s + N[u].cnt + Rank(N[u].ch[1],val);
	else return Rank(N[u].ch[0],val);
}

int Kth(int u,int k) {
	if (!u) return 0;
	if (k <= N[N[u].ch[0]].s) return Kth(N[u].ch[0],k);
	if (k > N[N[u].ch[0]].s + N[u].cnt) 
		return Kth(N[u].ch[1],k - N[N[u].ch[0]].s - N[u].cnt);
	return N[u].v;
}

int pred(int u, int val) {
	if (!u) return INF;
	if (val <= N[u].v) return pred(N[u].ch[0],val);
	int ans = pred(N[u].ch[1],val);
	if (ans == INF) ans = N[u].v;
	return ans;
}

int succ(int u, int val) {
	if (!u) return INF;
	if (val >= N[u].v) return succ(N[u].ch[1],val);
	int ans = succ(N[u].ch[0],val);
	if (ans == INF) ans = N[u].v;
	return ans;
}

int main() {
	//freopen("A.in","r",stdin);
	//freopen("A.out","w",stdout);
	int n, op, x;
	scanf("%d", &n);
	srand(time(NULL));
	while (n--) {
		scanf("%d%d", &op, &x);
		if (op == 1) Ins(root,x);
		else if (op == 2) Del(root,x);
		else if (op == 3) printf("%d\n", Rank(root,x));
		else if (op == 4) printf("%d\n", Kth(root,x));
		else if (op == 5) printf("%d\n", pred(root,x));
		else printf("%d\n",succ(root,x));
	}
	return 0;
}

附：在网上看见了一个模板，把splay，treap和SBT的做法打了出来。

传送门在这里