数据结构_我不会AVL_wbhavl

问题描述

　　欢迎来到暴走数据结构，我是洪尼玛。今天，我们来玩 AVL 树，怎么玩呢？ 很简单：给
你 n 个数字，你需要按顺序插入一棵 AVL树中，然后输出每个数所在节点的深度（从 1 开始）。
因为我不会 AVL 树，所以希望聪明的你们来帮我完成这个任务

★数据输入
输入第一个数为 n（n≤100000） 表示数字的个数
接下来一行输入 n 个数，范围在 1 到 n 之间，每个数只出现一次

★数据输出
按输入的顺序依次输出每个数所在节点的深度

输入示例	输出示例
6 1 2 3 4 5 6	3 2 3 1 2 3

★提示
注意： 输出行末不能有空格
对于 50%的数据， 1<=n<=100
对于 100%的数据， 1<=n<=100000

思路

　　没说明技巧，就是avl树

　　具体实现自行百度

code

 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>

 struct Node
 {
     int data;
     int height;
     int left;
     int right;
 };

 Node *arr = NULL;

 inline int Max(int a,int b) {return a>b?a:b;}
 inline int GetHeight(int root){
     return arr[root].height; //return root==0 ? 0 : arr[root]->height;
 }
 inline void CalcHeight(int root){
     arr[root].height = Max(GetHeight(arr[root].left),GetHeight(arr[root].right)) + ;
 }
 //------------------------------------------------------------------------------------
 int RightRotate(int root)
 {
     int newroot = arr[root].left;
     arr[root].left = arr[newroot].right;
     arr[newroot].right = root;
     CalcHeight(root);
     CalcHeight(newroot);
     return newroot;
 }
 int LeftRotate(int root)
 {
     int newroot = arr[root].right;
     arr[root].right = arr[newroot].left;
     arr[newroot].left = root;
     CalcHeight(root);
     CalcHeight(newroot);
     return newroot;
 }
 int LeftRightRotate(int root)
 {
     arr[root].left = LeftRotate(arr[root].left);
     return RightRotate(root);
 }
 int RightLeftRotate(int root)
 {
     arr[root].right = RightRotate(arr[root].right);
     return LeftRotate(root);
 }
 //------------------------------------------------------------------------------------
 int Insert(int root, int data)
 {
     if(root==)
     {
         arr[data].data = data;
         arr[data].height = ;
         return data;
     }
     else if(data < arr[root].data)
     {
         arr[root].left = Insert(arr[root].left,data);
         if(GetHeight(arr[root].left) - GetHeight(arr[root].right) >= )
         {
             if(data < arr[arr[root].left].data)
                 root = RightRotate(root);
             else//if(data > arr[arr[root]->left]->data)
                 root = LeftRightRotate(root);
         }
     }
     else if(data > arr[root].data)
     {
         arr[root].right = Insert(arr[root].right,data);
         if(GetHeight(arr[root].right) - GetHeight(arr[root].left) >= )
         {
             if(data > arr[arr[root].right].data)
                 root = LeftRotate(root);
             else//if(data < arr[arr[root]->right]->data)
                 root = RightLeftRotate(root);
         }
     }//else ==
     CalcHeight(root);
     return root;
 }

 void dfs(int index,int step)
 {
     if(index==) return;
     arr[index].height = step;
     dfs(arr[index].left,step+);
     dfs(arr[index].right,step+);
 }

 int main()
 {
     int n;
     scanf("%d",&n);
     int *data = (int *)malloc(sizeof(int)*(n+));
     arr = (Node *)malloc(sizeof(Node)*(n+));
     memset(arr,,sizeof(Node)*(n+));
     //------------------------------------------------
     int root = ;
     int i;
     for(i=; i<n; i++)
     {
         scanf("%d",data+i);
         root = Insert(root,data[i]);
     }
     dfs(root,);
     for(i=; i<n; i++)
     {
         if(i==n-)
             printf("%d\n",arr[data[i]].height);
         else
             printf("%d ",arr[data[i]].height);
     }

     //------------------------------------------------
     free(arr);
     free(data);
     return ;
 }