题目详情:https://www.patest.cn/contests/pat-a-practise/1066
提交:
代码:
#include <iostream>
#include <malloc.h>
using namespace std;
#define N 22
int n,number[N];
typedef struct AVL
{
int data,height; //height存储以当前节点为根节点的树的高度
struct AVL* leftChild;
struct AVL* rightChild;
}AVL;
int getHeight( AVL* root ) //求以root为根节点的树高
{
if( root == NULL )
return 0;
else
{
int leftChild = getHeight(root->leftChild);//得到左子树的树高
int rightChild = getHeight(root->rightChild);//得到右子树的树高
return leftChild>rightChild?leftChild+1:rightChild+1;//返回两者中的较大值
}
}
void updateHeight( AVL* root ) //更新以root为根节点的树的高度
{
root->height = getHeight(root);
}
int getHeightFactor( AVL* root ) //得到节点root的平衡因子
{
return getHeight(root->leftChild)-getHeight(root->rightChild);
}
void leftRotation( AVL* &root ) //左旋
{
AVL* temp = root->rightChild;
root->rightChild = temp->leftChild;
temp->leftChild = root;
updateHeight(root);
updateHeight(temp);
root = temp;
}
void rightRotation( AVL* &root ) //右旋
{
AVL* temp = root->leftChild;
root->leftChild = temp->rightChild;
temp->rightChild = root;
updateHeight(root);
updateHeight(temp);
root = temp;
}
AVL* insertAVL( AVL* root,int key ) //往平衡二叉树上插入节点
{
if( root == NULL ) //找到插入位置
{
root = (AVL*)malloc(sizeof(AVL));
root->data = key;
root->height = 1; //此时以该节点为根节点的树只有这一个节点,所以树高为1
root->leftChild = root->rightChild =NULL; //置空
return root;
}
else if( root->data > key ) //根节点的值较大,往左子树上查
{
root->leftChild = insertAVL(root->leftChild,key);
updateHeight(root); //更新节点的高度
if( getHeightFactor(root) == 2 ) //因为是往左子树上插,所以左子树的树高增大,平衡因子也是增大的
{
if( getHeightFactor(root->leftChild) == 1 ) //若root->leftChild的平衡因子为1,则是LL型树 ,右转即可
{
rightRotation(root);
}
else if( getHeightFactor(root->leftChild) == -1 ) //若root->leftChild的平衡因子为-1,则是LR型树 ,
{ //先左转(该节点的左子树左转)再右转
leftRotation( root->leftChild );
rightRotation(root);
}
}
return root;
}
else if( root->data < key ) //根节点的值较小,往右子树上查
{
root->rightChild = insertAVL(root->rightChild,key);
updateHeight(root); //更新节点的高度
if( getHeightFactor(root) == -2 ) //因为是往右子树上插,所以右子树的树高增大,平衡因子是减小的
{
if( getHeightFactor(root->rightChild) == -1 ) //若root->leftChild的平衡因子为-1,则是RR型树 ,左转即可
{
leftRotation(root);
}
else if( getHeightFactor(root->rightChild) == 1 ) //若root->rightChild的平衡因子为1,则是RL型树 ,
{ //先右转(该节点的右子树左转)再左转
rightRotation( root->rightChild );
leftRotation(root);
}
}
return root;
}
}
int main ()
{
cin>>n;
AVL* root = NULL;
for( int i=0;i<n;++i ) //处理输入
{
cin>>number[i];
root = insertAVL(root,number[i]);
}
cout<<root->data<<endl;
return 0;
}
我参考着《算法笔记》写出来的,至于为什么节点的平衡因子不会超过[-2,2]这个范围,个人猜想插入一个节点,平衡因子只能渐变为2或者-2,而一旦发现平衡因子为2或者-2,当即调整,所以不会超出这个范围。又可以根据孩子节点的平衡因子值判断是什么样的树形,进行相应的调整。
以前以为平衡二叉树挺难得,今天敲出来感觉也就这样 [手动傲娇状].另外有些程序很相似,比如上面的左旋,右旋与交换两个数的函数(如下):
交换两个数
void swap ( int &a,int &b )
{
int temp = a;
a = b;
b = temp;
}
左旋:
void leftRotation( AVL* &root ) //左旋
{
AVL* temp = root->rightChild;
root->rightChild = temp->leftChild;
temp->leftChild = root;
updateHeight(root);
updateHeight(temp);
root = temp;
}
右旋:
void rightRotation( AVL* &root ) //右旋
{
AVL* temp = root->leftChild;
root->leftChild = temp->rightChild;
temp->rightChild = root;
updateHeight(root);
updateHeight(temp);
root = temp;
}
发现没?都是首尾相接的,有点对称的感觉。
在这里借用这篇博客中的图: http://blog.csdn.net/collonn/article/details/20128205
情况1和情况2都是LL型树,需要进行右转。
情况3和情况4都是RR型树,需要进行左转。
情况5和情况6都是LR型树,先失衡节点的左子树左转,转换成LL型后再右转。
情况5和情况6都是RL型树,先失衡节点的右子树右转,转换成RR型后再左转。
AVL树在插入结束后,递归返回的过程检查各个节点的平衡因子,并调整。而插入的过程和BST树基本相同。