数据结构_我不会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)) + 1;
}
//------------------------------------------------------------------------------------
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==0)
    {
        arr[data].data = data;
        arr[data].height = 1;
        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) >= 2)
        {
            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) >= 2)
        {
            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==0) return;
    arr[index].height = step;
    dfs(arr[index].left,step+1);
    dfs(arr[index].right,step+1);
}

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