splay板子

解锁新技能,伸展树,splay基于平衡二叉树,可以查询前驱和后继,保证差值最小哦

以x为主人公

右旋
无图(自行脑补qaq)

y节点为x节点的父节点,z节点为y节点的父节点(前置)
1.如果x节点存在右节点,那么就把它作为y节点的左儿子,更新x的右儿子的父亲为y节点。
void right_rotate(int x){
  int y=fa[x];
  int z=fa[y];
  fa[rightson[x]]=y;
  if (rightson[x]) {
       leftson[y]=rigthson[x]; 
      }

2.如果存在z且z为y的左儿子,那么x就为y的左儿子,更新
 if(z) {
   if (y==leftson[z]) leftson[z]=x;
   else rightson[z]=x;//这个地方对照一下操错了QAQ
 }
3.将y作为x的右子树
rightson[x]=y;
fa[y]=x;
}

左旋操作

left_rotate(int x)
{
  int y=fa[x];
  int z=fa[y];
  fa[leftson[x]]=y;
  if (leftson[x]){
     rightson[y]=fa[leftson[x]];
  }
  if (z){
    if (rightson[z]==y) rightson[z]=x;
    else leftson[z]=x;
  }
  //将y作为x的左子树
leftson[x]=y;
fa[y]=x;
}

差不多辽,接下来上splay


void splay(int ancestry){
	while (fa[x]!=ancestry){
		int y=fa[x];
		int z=fa[y];
		if (z==ancestry){
			if (rightson[y]==x) left_rotate(x);
			else right_rotate(x);
		}
		else {
			if (rightson[z]==y&&rightson[y]==x) {
				left_rotate(y);left_rotate(x);
			}
		   else if (leftson[z]==y&&leftson[y]==x)
		       {
		       	right_rotate(y);right_rotate(x);
			   }
		   else if (leftson[z]==y&&rightson[y]==x) 
		        {
		        	right_rotate(y);
		        	left_rotate(x);
				}
			else if (rightson[z]==y&&leftson[y]==x)
			  {
			  	  left_rotate(y);
			  	  right_rotate(x);
			  }
		}
	}
	if (ancestry==0) root=x;
}
全部评论

相关推荐

不愿透露姓名的神秘牛友
11-20 19:57
已编辑
某大厂 golang工程师 23.0k*16.0, 2k房补,年终大概率能拿到
点赞 评论 收藏
分享
Noel_:中石油是这样的 哥们侥幸混进免笔试名单 一看给我吓尿了
点赞 评论 收藏
分享
10-05 11:11
海南大学 Java
投票
理想江南137:感觉挺真诚的 感觉可以试一试
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务