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

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

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

DQE:

结点平衡因子改为该结点树深度，旋转操作记得更新结点深度。

 

 #include <iostream>
 #include <cstdio>
 using namespace std;
 /*
 全测数据
 in
 9
 16 3 7 11 9 26 18 14 15
 out
 11 7 3 9 18 15 14 16 26
 3 7 9 11 14 15 16 18 26
 */
 struct BT
 {
     int d;
     int x;
     BT *lt,*rt;
 };

 int dp(BT *r)
 {
     return r?r->d:;
 }
 void rfd(BT *r)
 {
     int ld=dp(r->lt)+,rd=dp(r->rt)+;
     r->d=ld>rd?ld:rd;
 }

 void LL(BT *&r)
 {
 //    printf("LL#%d#%d*%d ",r->x,dp(r->lt),dp(r->rt));
     BT *a=r,*b=a->lt,*br=b->rt;
     r=b;    //利用引用修改上级指向
     b->rt=a;
     a->lt=br;
     //深度重定
     rfd(a);
 }
 void RR(BT *&r)
 {
 //    printf("RR#%d#%d*%d ",r->x,dp(r->rt),dp(r->lt));
     BT *a=r,*b=a->rt,*bl=b->lt;
     r=b;
     b->lt=a;
     a->rt=bl;

     rfd(a);
 }
 void LR(BT *&r)
 {
     RR(r->lt);
     rfd(r->lt);
     LL(r);
 }
 void RL(BT *&r)
 {
     LL(r->rt);
     rfd(r->rt);
     RR(r);
 }

 void insert(BT *&root,int e)
 {
     if(!root)
     {
         BT *r=new BT;
         r->x=e;
         r->d=;
         r->lt=r->rt=NULL;
         root=r;
     }
     else
     {
         if(e<root->x)
         {
             insert(root->lt,e);
             if(dp(root->lt)-dp(root->rt)>)
             {
                 if(e<root->lt->x)
                     LL(root);
                 else
                     LR(root);
             }
         }
         else if(e>root->x)
         {
             insert(root->rt,e);
             if(dp(root->rt)-dp(root->lt)>)
             {
                 if(e>root->rt->x)
                     RR(root);
                 else
                     RL(root);
             }
         }
         //更新根深度
         rfd(root);
     }
 }

 void xout(BT *r)
 {if(r){printf("%d ",r->x);xout(r->lt);xout(r->rt);}}
 void mout(BT *r)
 {if(r){mout(r->lt);printf("%d ",r->x);mout(r->rt);}}
 int main()
 {
     BT *root=NULL;
     int n,i;
     scanf("%d",&n);
     while(n--)
     {
         scanf("%d",&i);
         insert(root,i);
     }
     /*
     putchar('\n');
     xout(root);
     putchar('\n');
     mout(root);
     */
     printf("%d\n",root->x);
     return ;
 }

 /***************************************************
 User name: ***
 Result: Accepted
 Take time: 0ms
 Take Memory: 156KB
 Submit time: 2016-11-29 19:09:25
 ****************************************************/