题目
https://pintia.cn/problem-sets/994805342720868352/problems/994805404939173888
AC代码
#include<bits/stdc++.h>
using namespace std;
typedef struct node* avl;
struct node{
int data;
avl left,right;
};
int GetHeight(avl a)//获取高度
{
if(!a) return 0;
else
{
int l=GetHeight(a->left)+1;
int r=GetHeight(a->right)+1;
return l>r?l:r;
}
}
avl L(avl a)//左旋转
{
avl b=a->right;
a->right=b->left;
b->left=a;
return b;
}
avl R(avl a)//右旋转
{
avl b=a->left;
a->left=b->right;
b->right=a;
return b;
}
avl LR(avl a)//左右旋转
{
a->left=L(a->left);
return R(a);
}
avl RL(avl a)//右左旋转
{
a->right=R(a->right);
return L(a);
}
avl Insert(avl a,int tmp)//插入节点
{
if(!a)
{
avl a=(avl)malloc(sizeof(struct node));//必须要为a开辟空间
a->data=tmp;
a->left=a->right=NULL;
return a;
//这里的return a一定要写,因为新开辟的a是局部变量,在if判断后就被剥夺空间了,所以必须在它的作用域内返回
}
else if(tmp<a->data)//平衡二叉树是二叉排序树,左小右大
{
a->left=Insert(a->left,tmp);
if(GetHeight(a->left)-GetHeight(a->right)==2)
{
if(tmp<a->left->data) a=R(a);
else a=LR(a);
}
}
else //平衡二叉树是二叉排序树,左小右大
{
a->right=Insert(a->right,tmp);
if(GetHeight(a->left)-GetHeight(a->right)==-2)
{
if(tmp>a->right->data) a=L(a);
else a=RL(a);
}
}
return a;
}
int main()
{
int n,tmp;
cin>>n;
avl a;//这个NULL的赋值也要写,否则会出现段错误,因为要为a开辟空间
while(n--)
{
cin>>tmp;
a=Insert(a,tmp);
}
cout<<a->data;
}
说明
建议不了解AVL是什么的同学先看看这篇文章——漫话:什么是平衡(AVL)树?这应该是把AVL树讲的最好的文章了,主要看数据结构的讲解部分,代码看我这里的就行
当你了解之后再结合刚才的理论看左旋,右旋,左右旋,右左旋对应的代码,相信能更好地帮助你理解
tips:在理解过程中适当作图是一个不错的选择