【整合】SBT模板(全部操作)

插入,删除,前驱后继,第k大值,排名,最大值最小值8种询问外带核心左旋右旋Maintain

全套模板整合

从16:10开始写,写到了16:28

总计用时18min左右(前后误差不超过1min)

我的手速还真是快呢owo

这一次就是个小测试了。看来到现在SBT已经写熟练了0-0

可以去颓Splay和fhqTreap之类的了www

P.S.应该没写错吧。。。有什么写错了的有人看出来了马上告诉我QUQ

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
struct sbt
{
	int left,right,size,data;
}tree[100000];
int top,root;
int n;
int opt;
int key;
void left_rot(int &x)
{
	int y=tree[x].right;
	tree[x].right=tree[y].left;
	tree[y].left=x;
	tree[y].size=tree[x].size;
	tree[x].size=tree[tree[x].left].size+tree[tree[x].right].size+1;
	x=y;
}
void right_rot(int &x)
{
	int y=tree[x].left;
	tree[x].left=tree[y].right;
	tree[y].right=x;
	tree[y].size=tree[x].size;
	tree[x].size=tree[tree[x].left].size+tree[tree[x].right].size+1;
	x=y;
}
void Maintain(int &x,bool flag)
{
	if (!flag)
	{
		if (tree[tree[tree[x].left].left].size>tree[tree[x].right].size)
		    right_rot(x);
		else
		if (tree[tree[tree[x].left].right].size>tree[tree[x].right].size)
		    left_rot(tree[x].left),right_rot(x);
		else
		return;
	}
	else
	{
		if (tree[tree[tree[x].right].right].size>tree[tree[x].left].size)
		    left_rot(x);
		else
		if (tree[tree[tree[x].right].left].size>tree[tree[x].left].size)
		    right_rot(tree[x].right),left_rot(x);
		else
		return;
	}
	Maintain(tree[x].left,0);
	Maintain(tree[x].right,1);
	Maintain(x,1);
	Maintain(x,0);
}
void insert(int &x,int data)
{
	if (x==0)
	{
		x=++top;
		tree[x].size=1;
		tree[x].left=tree[x].right=0;
		tree[x].data=data;
	}
	else
	{
		tree[x].size++;
		if (data<tree[x].data) insert(tree[x].left,data);
		else insert(tree[x].right,data);
		Maintain(x,data>=tree[x].data);
	}
}
int remove(int &x,int data)
{
	tree[x].size--;
	if (data>tree[x].data)
	    remove(tree[x].right,data);
	else
	if (data<tree[x].data)
	    remove(tree[x].left,data);
	else
	{
		if (tree[x].left!=0&&tree[x].right==0)
		{
			int ret=x;
			x=tree[x].left;
			return x;
		}
		else
		if (tree[x].right!=0&&tree[x].left==0)
		{
			int ret=x;
			x=tree[x].right;
			return x;
		}
		else
		if (tree[x].left==0&&tree[x].right==0)
		{
			int ret=x;
			x=0;
			return ret;
		}
		else
		{
			int ret=tree[x].right;
			while (tree[ret].left) ret=tree[x].left;
			tree[x].data=tree[ret].data;
			remove(tree[x].right,tree[ret].data);
		}
	}
}
int select(int &x,int k)
{
	int r=tree[tree[x].left].size+1;
	if (k<r) return select(tree[x].left,k);
	else
	if (k>r) return select(tree[x].right,k-r);
	else
	return tree[x].data;
}
int rank(int &x,int data)
{
	if (x==0) return 0;
	if (data>tree[x].data) return tree[tree[x].left].size+1+rank(tree[x].right,data);
	else
	if (data<tree[x].data) return rank(tree[x].left,data);
	else
	return tree[tree[x].left].size+1;
}
int prev(int &x,int y,int data)
{
	if (x==0) return y;
	if (data>=tree[x].data) return prev(tree[x].right,x,data);
	else return prev(tree[x].left,y,data);
}
int succ(int &x,int y,int data)
{
	if (x==0) return y;
	if (data>=tree[x].data) return succ(tree[x].right,y,data);
	else return succ(tree[x].left,x,data);
}
int getmin()
{
	int x=root;
	while (tree[x].left) x=tree[x].left;
	return tree[x].data;
}
int getmax()
{
	int x=root;
	while (tree[x].right) x=tree[x].right;
	return tree[x].data;
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&opt);
		switch (opt)
		{
			case 1:
				scanf("%d",&key);
				insert(root,key);
				break;
			case 2:
				scanf("%d",&key);
				printf("%d\n",rank(root,key));
				break;
			case 3:
				scanf("%d",&key);
				printf("%d\n",select(root,key));
				break;
			case 4:
				printf("%d\n",getmax());
				break;
			case 5:
				printf("%d\n",getmin());
				break;
			case 6:
				scanf("%d",&key);
				printf("%d\n",succ(root,0,key));
				break;
			case 7:
				scanf("%d",&key);
				printf("%d\n",prev(root,0,key));
				break;
			case 8:
				scanf("%d",&key);
				remove(root,key);
		}
	}
}


评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值