数据结构实验之查找二：平衡二叉树

数据结构实验之查找二：平衡二叉树

Time Limit: 400MS Memory Limit: 65536KB
Submit Statistic
Problem Description

根据给定的输入序列建立一棵平衡二叉树，求出建立的平衡二叉树的树根。
Input

输入一组测试数据。数据的第1行给出一个正整数N(n <= 20)，N表示输入序列的元素个数；第2行给出N个正整数，按数据给定顺序建立平衡二叉树。
Output

输出平衡二叉树的树根。
Example Input
5
88 70 61 96 120

Example Output
70

Hint

Author

xam

平衡二叉树动态演示视频：http://www.icoolxue.com/play/3178
平衡二叉树代码实现视频：http://www.icoolxue.com/play/3179
很经典！

AC Code：

#include<iostream>
using namespace std;
struct node
{
    int d;
    node *l,*r;
    int deep;
};
node *root;
int deept(node *p)      //计算树的深度
{
    if(!p)
        return -1;
    return p->deep;
}
void LL(node *&p)//左旋
{
    node *q=p->l;
    p->l=q->r;
    q->r=p;
    p->deep=max(deept(p->r),deept(p->l))+1;
    q->deep=max(p->deep,deept(q->l))+1;
    p=q;            //将p和q交换
}
void RR(node *&p)//右旋
{
    node *q=p->r;
    p->r=q->l;
    q->l=p;
    p->deep=max(deept(p->l),deept(p->r))+1;
    q->deep=max(deept(q->r),p->deep)+1;
    p=q;
}
void LR(node *&p)
{
    RR(p->l);
    LL(p);
}
void RL(node *&p)
{
    LL(p->r);
    RR(p);
}
void InsertT(node *&p,int k)
{
    if(!p)          //如果节点为空,就在此节点处加入k信息
    {
        p=new node;
        p->d=k;
        p->l=p->r=NULL;
        p->deep=1;
    }
        else if(k<p->d)   //如果k小于节点的值,就继续在节点的左子树中插入k
    {
        InsertT(p->l,k);
        if(deept(p->l)-deept(p->r)>1)//如果高度之差大于1的话就失去了平衡,需要旋转
        {
            if(k<p->l->d)
                LL(p);
            else
                LR(p);
        }
    }
    else if(k>p->d)
    {
        InsertT(p->r,k);
        if(deept(p->r)-deept(p->l)>1)
        {
            if(k>p->r->d)
                RR(p);
            else RL(p);
        }
    }
    p->deep=max(deept(p->l),deept(p->r))+1;
}

int main()
{
    int n;
    cin>>n;
    root=NULL;
    while(n--)
    {
        int x;
        cin>>x;
        InsertT(root,x);
    }
    cout<<root->d<<endl;
    return 0;
}