【洛谷 P3369】 【平衡树】 普通平衡树

【洛谷 P3369】 【平衡树】 普通平衡树

题目

---

解题思路

先认识一个东西叫BST（二叉查找树）

• 左子树上的关键值一定比该节点的小
• 右子树上的关键值一定比该节点大

平衡树就是基于BST的基础上，改变它的形态使左右子树大小达到平衡
这种改变形态并保持BST性质的操作叫**旋转 **
本人有点蠢，为了好理解，定义左旋为将左儿子变为爸爸，右旋为将右儿子变为爸爸

具体操作看代码
此操作方便删点时，让点变为叶子节点，那对树的影响很小

题目上的六种操作代码里有注释

---

代码

#include<iostream>
#include<stdlib.h> 
#include<cstdio>
const int inf=2147483647;
using namespace std;
struct lzf{
	int val,l,r,dat,cnt,size;
}a[100010];
int n,x,y,tot,root;
int New(int val)  //新建节点
{
	 a[++tot].val=val;
	 a[tot].dat=rand();  //随机优先级（奇奇怪怪的东西，好像是方便旋转）
	 a[tot].cnt=a[tot].size=1;  //cnt是这个val出现过多少次，size是以ta为根的子树大小
	 return tot;
}
void qsize(int p)  //求size大小
{ 
	 a[p].size=a[a[p].l].size+a[a[p].r].size+a[p].cnt;
} 
void zuo(int &p)  //左旋
{
	 int q=a[p].l;
	 a[p].l=a[q].r,a[q].r=p;
	 p=q;
	 qsize(a[p].r),qsize(p);
}
void you(int &p)  //右旋
{
	int q=a[p].r;
	a[p].r=a[q].l,a[q].l=p;
	p=q;
	qsize(a[p].l),qsize(p);
}
void insert(int &p,int val)  //插入一个新值
{ 
	 if (!p)  //这个位置为空，插入的值可以放入
	 {
	 	p=New(val); 
	 	return;
	 }
	 if (val==a[p].val)  //这个值已经出现过
	 {
	 	a[p].cnt++;
	 	a[p].size++;
	 	return;
	 }
	 if (val<a[p].val)  //继续往下传递，找空位
	 {
	 	insert(a[p].l,val);
	 	if (a[p].dat<a[a[p].l].dat) zuo(p);
	 }
	 else {
	 	insert(a[p].r,val);
	 	if (a[p].dat>a[a[p].r].dat) you(p);
	 }
	 qsize(p);
}
void cut(int &p,int val)  //删除一个值
{
	 if (p==0) return;  
	 if (val==a[p].val)
	 {
	 	if (a[p].cnt>1)  //出现很多次，减1次，不删点
	 	{
	 	   a[p].cnt--,a[p].size--;
	 	   return;
	    }
	    if (a[p].l||a[p].r)  //不是叶子节点，将点往下旋再删去
	    { 
	    	if (!a[p].r||a[a[p].l].dat>a[a[p].r].dat)  //没有右子树，或左子树优先级更大就左旋，不然右旋
	    	   zuo(p),cut(a[p].r,val);
	    	   else you(p),cut(a[p].l,val);
	    	qsize(p);
		} 
		else p=0;
		return;
	}
	if (val<a[p].val)  //继续找这个值
	   cut(a[p].l,val);
	   else cut(a[p].r,val);
	qsize(p);
} 
int ask_rank(int p,int x)  //找到排名为x的值
{
	if (p==0) return inf;
	if (a[a[p].l].size>=x) return ask_rank(a[p].l,x);  
	if (a[a[p].l].size+a[p].cnt>=x) return a[p].val; 
	return ask_rank(a[p].r,x-a[a[p].l].size-a[p].cnt);
}
int ask_val(int p,int x)  //找到值为x的点的排名
{
	if (p==0) return 0;
	if (a[p].val==x) return a[a[p].l].size+1;
	if (x<a[p].val) return ask_val(a[p].l,x);
	return ask_val(a[p].r,x)+a[a[p].l].size+a[p].cnt; 
}
int ask_bf(int x)  //求前驱，就是找比x小中最大的那个数
{
	int w=1,p=root;
	while (p)
	{
		  if (x==a[p].val)
		  {
		     if (a[p].l)
		     {
		     	p=a[p].l;
		     	while (a[p].r) p=a[p].r;
		     	w=p;
			 }
			 break;
		  }
		  if (x>a[p].val&&a[p].val>a[w].val) w=p;
		  if (x<a[p].val)
		     p=a[p].l;
		     else p=a[p].r;
	}
	return a[w].val;
}
int ask_nx(int x)  //求后继，就是找比x大中最小的那个数
{
	int w=2,p=root;
	while (p)
	{
		  if (x==a[p].val)
		  {
		     if (a[p].r)
		     {
		     	p=a[p].r;
		     	while (a[p].l) p=a[p].l;
		     	w=p;
			 }
			 break;
		  } 
		  if (x<a[p].val&&a[p].val<a[w].val) w=p;
		  if (x<a[p].val)
		     p=a[p].l;
		     else p=a[p].r;
	}
	return a[w].val;
}
int main()
{
	scanf("%d",&n);
	root=New(-inf);a[1].r=New(inf);
	qsize(root);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",&x,&y);
		if (x==1) insert(root,y);
		if (x==2) cut(root,y);
		if (x==3) printf("%d\n",ask_val(root,y)-1);
		if (x==4) printf("%d\n",ask_rank(root,y+1));
		if (x==5) printf("%d\n",ask_bf(y));
		if (x==6) printf("%d\n",ask_nx(y));
	}
	return 0;
}