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

最新推荐文章于 2021-07-22 22:41:02 发布

原创最新推荐文章于 2021-07-22 22:41:02 发布 · 443 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#山东理工ACM #数据结构实验

本文介绍了一种根据给定序列构建平衡二叉树的方法，并提供了完整的C++代码实现。通过该方法，可以有效地找到构建后的平衡二叉树的根节点。

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

Time Limit: 400MS Memory limit: 65536K

题目描述

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

输入

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

输出

输出平衡二叉树的树根。

示例输入

5
88 70 61 96 120

示例输出

提示

#include<bits/stdc++.h>
using namespace std;
struct node
{
    int data;
    int high;
    struct node *l;
    struct node *r;
};
int max(int x,int y)
{
    if(x>y)
    {
        return x;
    }
    else
        return y;
}
int deep(struct node *T)
{
    if(!T)
        return -1;
    else
        return T->high;
}
struct node *ll(struct node *root)
{
    struct node *p;
    p=root->l;
    root->l=p->r;
    p->r=root;
    p->high=max(deep(p->l),deep(p->r))+1;
    root->high=max(deep(root->l),deep(root->r))+1;
    return p;
};
struct node *rr(struct node *root)
{
    struct node *p;
    p=root->r;
    root->r=p->l;
    p->l=root;
    p->high=max(deep(p->l),deep(p->r))+1;
    root->high=max(deep(root->l),deep(root->r))+1;
    return p;
};
struct node *rl(struct node *root)
{
    root->l=rr(root->l);
    return ll(root);
};
struct node *lr(struct node *root)
{
    root->r=ll(root->r);
    return rr(root);
};
struct node *creat(struct node *root,int e)
{
    if(!root)
    {
        root=new node;
        root->data=e;
        root->high=0;
        root->l=NULL;
        root->r=NULL;
    }
    else if(root->data>e)
    {
        root->l=creat(root->l,e);
        if(deep(root->l)-deep(root->r)>1)
        {
            if(e<root->l->data)
                root=ll(root);
            else
                root=rl(root);
        }
    }
    else if(root->data<e)
    {
        root->r=creat(root->r,e);
        if(deep(root->r)-deep(root->l)>1)
        {
            if(e>root->r->data)
            {
                root=rr(root);
            }
            else
                root=lr(root);
        }
    }
    root->high=max(deep(root->l),deep(root->r))+1;
    return root;
};
int main()
{
    int n;
    int num;
    cin>>n;
    struct node *root;
    root=NULL;
    for(int i=0;i<n;i++)
    {
        cin>>num;
        root=creat(root,num);
    }
    cout<<root->data<<endl;
    return 0;
}