欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

平衡二叉树(AVL)

程序员文章站 2023-03-31 13:02:56
AVL就是优化二叉查找树 平衡因子不大于1 左 < 根 < 右 具体看代码 ......

avl就是优化二叉查找树

平衡因子不大于1

左 < 根 < 右

 

具体看代码

#include<bits/stdc++.h>

using namespace std;
typedef struct node;
typedef node * tree;
struct node
{
    int v;
    int heigh;
    tree l,r;
};

//获取以root为根结点的子树的当前height
int getheigh(tree root)
{
    if(root==null) return 0;
    return root->heigh;
}

//更新结点root的heigh
void updataheigh(tree root)
{
    //max(左孩子结点的height,有孩子结点的height)+1
    root->heigh=max(getheigh(root->l),getheigh(root->r))+1;
}


//计算平衡因子
int getbalance(tree root)
{
    //左-右
    return getheigh(root->l)-getheigh(root->r);
}


//左旋  注意原理 对于rr是root的右孩子的平衡因子是-1
void l(tree &root)
{
    tree temp;
    temp=root->r;
    root->r=temp->l;
    temp->l=root;
    updataheigh(root);
    updataheigh(temp);
    root=temp;
}

void r(tree &root)
{
    tree temp;
    temp=root->l;
    root->l=temp->r;
    temp->r=root;
    updataheigh(root);
    updataheigh(temp);
    root=temp;
}
void insertt(tree &root,int v)
{
    if(root==null){//当结点是空的时候 就是插入的时候
        root=new node;
        root->v=v;
        root->heigh=1;
        root->l=root->r=null;
        return;
    }
    if(v<root->v){
        insertt(root->l,v);
        updataheigh(root);//注意更新树高
        if(getbalance(root)==2){
            if(getbalance(root->l)==1){
                r(root);
            }
            else if(getbalance(root->l)==-1){
                l(root->l);
                r(root);
            }
        }
    }
    else{
        insertt(root->r,v);
        updataheigh(root);
        if(getbalance(root)==-2){
            if(getbalance(root->r)==-1){
                l(root);
            }
            else if(getbalance(root->r)==1){
                r(root->r);
                l(root);
            }
        }
    }

}
int main()
{
    int n;
    scanf("%d",&n);
    int x;
    tree root;
    root=null;
    for(int i=0;i<n;i++){
        scanf("%d",&x);
        insertt(root,x);
    }
    printf("%d\n",root->v);
    return 0;
}