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

最新推荐文章于 2020-03-28 15:56:44 发布

原创最新推荐文章于 2020-03-28 15:56:44 发布 · 433 阅读

0 ·

CC 4.0 BY-SA版权

树与二叉树专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种自平衡二叉搜索树——AVL树的实现方法，通过旋转操作来确保树的高度平衡，从而保持良好的查找性能。具体涵盖了左旋、右旋以及左右旋、右左旋等操作，并实现了节点插入后的平衡调整。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct bst {
    int d;
    bst *l,*r;
    int deep;
    bst() {
        l=r=NULL;
        deep=0;
    }
};
int deept(bst *p) {
    if(!p)
        return -1;
    return p->deep;
}
bst *LL(bst *p) {
    bst *q=p->l;
    p->l=q->r;
    q->r=p;
    p->deep=max(deept(p->r),deept(p->l))+1;
    q->deep=max(p->deep,deept(q->l))+1;
    return q;
}
bst *RR(bst *p) {
    bst *q=p->r;
    p->r=q->l;
    q->l=p;
    p->deep=max(deept(p->l),deept(p->r))+1;
    q->deep=max(deept(q->r),p->deep)+1;
    return q;
}
bst *LR(bst *p) {
    p->l=RR(p->l);
    return LL(p);
}
bst *RL(bst *p) {
    p->r=LL(p->r);
    return RR(p);
}
bst *InsertT(bst *p,int k) {
    if(!p) {
        p=new bst;
        p->d=k;
    }
    else if(k<p->d) {
        p->l=InsertT(p->l,k);
        if(deept(p->l)-deept(p->r)>1) {
            if(k<p->l->d)
                p=LL(p);
            else
                p=LR(p);
        }
    }
    else if(k>p->d) {
        p->r=InsertT(p->r,k);
        if(deept(p->r)-deept(p->l)>1) {
            if(k>p->r->d)
                p=RR(p);
            else p=RL(p);
        }
    }
    p->deep=max(deept(p->l),deept(p->r))+1;
    return p;
}

int main() {
    int n;
    scanf("%d", &n);
    bst *root=NULL;
    while(n--) {
        int x;
        scanf("%d", &x);
        root=InsertT(root,x);
    }
    printf("%d", root->d);
    return 0;
}