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

数据结构实验之查找二：平衡二叉树 （ LL RR LR RL）

Problem Description

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

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

输出平衡二叉树的树根。
Sample Input

5
88 70 61 96 120
Sample Output

70

#include <iostream>
#include <bits/stdc++.h>

using namespace std;

struct node
{
    node *ltree;
    node *rtree;
    int dp;
    int data;
};

int deep(node *p)
{
    if(p==NULL)
    {
        return -1;
    }
    else
    {
        return p->dp;
    }
}

node *LL(node *root)//注意是头相右旋转
{
    node *p=root->ltree;
    root->ltree=p->rtree;
    p->rtree=root;
    p->dp=max(deep(p->ltree),root->dp)+1;//p的右就是root   p和root的深度改变了
    root->dp=max(deep(root->ltree),deep(root->rtree))+1;//头变成了p
    return p;
}

node *RR(node *root)
{
    node *p=root->rtree;
    root->rtree=p->ltree;
    p->ltree=root;
    p->dp=max(deep(p->rtree),root->dp)+1;
    root->dp=max(deep(root->ltree),deep(root->rtree))+1;
    return p;
}

node *LR(node *root)
{
    root->ltree=RR(root->ltree);
    return LL(root);
}

node *RL(node *root)
{
    root->rtree=LL(root->rtree);
    return RR(root);
}

node *creat(node *root,int x)
{
    if(root==NULL)
    {
        root= new node;
        root->ltree=NULL;
        root->rtree=NULL;
        root->data=x;
        root->dp=0;//他的l和r是-1 他是0
    }
    else if(x<root->data)
    {
        root->ltree=creat(root->ltree,x);
        if(deep(root->ltree)-deep(root->rtree)>1)//因为插在左边肯定左边的增大
        {
            if(x<root->ltree->data)//再判断是在左边的左边还是左边的右边
            {
                root=LL(root);
            }
            else
            {
                root=LR(root);
            }
        }
    }
    else if(x>root->data)
    {
        root->rtree=creat(root->rtree,x);
        if(deep(root->rtree)-deep(root->ltree)>1)
        {
            if(x>root->rtree->data)
            {
                root=RR(root);
            }
            else
            {
                root=RL(root);
            }
        }
    }
    root->dp=max(deep(root->ltree),deep(root->rtree))+1;//求一下深度
    return root;
}

int main()
{
    int a,x;
    node *root=NULL;
    cin >> a;
    while(a--)
    {
        cin >> x;
        root=creat(root,x);
    }
    cout << root->data;
    return 0;
}