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

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

AC_Code

#include <stdio.h>
#include <stdlib.h>
struct node{
    int data,d;
    struct node *l,*r;
};

int max(int x,int y){
    if( x>y ) return x;
    else return y;
}

int deep(struct node *root){
    if(root==NULL){
        return -1;
    }
    else return root->d;
}

struct node *rr(struct node *root){
    struct node *p;
    p = root->r;
    root->r = p->l;
    p->l = root;
    root->d = max(deep(root->l),deep(root->r))+1;
    p->d = max(deep(p->l),deep(p->r))+1;
    return p;
}

struct node *ll(struct node *root){
    struct node *p;
    p = root->l;
    root->l = p->r;
    p->r = root;
    root->d = max(deep(root->l),deep(root->r))+1;
    p->d = max(deep(p->l),deep(p->r))+1;
    return p;
}

struct node *lr(struct node *root){
    root->l = rr(root->l);
    root = ll(root);
    return root;
}

struct node *rl(struct node *root){
    root->r = ll(root->r);
    root = rr(root);
    return root;
}

struct node *creat(int x, struct node *root){
    if( root==NULL ){
        root=(struct node *)malloc(sizeof(struct node));
        root->d=0;
        root->data=x;
        root->l=NULL;
        root->r=NULL;
    }
    else{
        if( x<root->data ){
            root->l=creat(x,root->l);
            if( deep(root->l)-deep(root->r)>1 ){
                if( x<root->l->data){
                    root=ll(root);
                }
                else root=lr(root);
            }
        }
        else{
            root->r=creat(x,root->r);
            if( deep(root->r)-deep(root->l)>1){
                if( x>root->r->data ){
                    root=rr(root);
                }
                else root=rl(root);
            }
        }
    }
    root->d=max(deep(root->l),deep(root->r))+1;
    return root;
}

int main()
{
    int x,i,n;
    scanf("%d",&n);
    struct node *root;
    root=NULL;

    for(i=1;i<=n;i++){
        scanf("%d",&x);
        root=creat(x,root);
    }
    printf("%d
", root->data);
    return 0;
}