数据结构 04-树5 Root of AVL Tree (25 分)

#include <iostream>

using namespace std;
typedef int ElementType;
class tnode{
public:
    ElementType data;
    tnode* left{nullptr};
    tnode* right{nullptr};
    tnode()=default;
    tnode(int d):data{d}{};
};

class AVLtree {
public:
    int size{0};
    tnode* root{nullptr};
    AVLtree():root{nullptr}{
    }
    
    tnode* initiation(){
        int n,temp;
        cin >> n;
        for(int i=0;i<n;i++){
            cin >> temp;
            insertNode(temp);
        }
        return root;
    }
    
    int getHeight(tnode* p){
        if(!p){
            return -1;
        }
        return max(getHeight(p->left),getHeight(p->right))+1;
    }
    
    int getBalanceFactor(tnode* p){
        if(!p){
            return 0;
        }
        return getHeight(p->left)-getHeight(p->right);
    }
    
    //单左旋
    tnode* leftRotate(tnode* p){
        tnode* y=p->right;
        p->right=y->left;
        y->left=p;
        return y;
    }
    
    //单右旋
    tnode* rightRotate(tnode* p){
        tnode* y=p->left;
        p->left=y->right;
        y->right=p;
        return y;
    }
    tnode* insertNode(tnode* p,const int &n){
        if(!p){
           p = new tnode{n};
        }else if(n<p->data){
            p->left=insertNode(p->left, n);
        }else if(n>p->data){
            p->right=insertNode(p->right, n);
        }else{
            return p;
        }
        int balanceFactor=getBalanceFactor(p);
        if(balanceFactor>1){//左子树高-右子树高大于1
            //左左 右旋
            if(n<p->left->data){
                return rightRotate(p);
            }else if(n>p->left->data){
                //左右 先左旋 再右旋
                p->left= leftRotate(p->left);
                return rightRotate(p);
            }
        }else if(balanceFactor<-1){
            if(n>p->right->data){
                return leftRotate(p);
            }else if(n<p->right->data){
                p->right=rightRotate(p->right);
                return leftRotate(p);
            }
        }
        return p;
    }
    void insertNode(const int &n){
        root=insertNode(root,n);
    }
    void deleteNode(){
        
    }
    void searchNode(){
        
    }
    
    

    //先左旋再右旋
    tnode* LeftrightRotate(tnode* p){
        return p;
    }
    //先右旋再左旋
    tnode* rightLeftRotate(tnode* p){
        return p;
    }
    
    void preOrderTraversal(){
        
    }
    void inOrderTraversal(){
        
    }
    void postOrderTraversal(){
        
    }

    void getMaxNode(){
        
    }
    void getMinNode(){
        
    }
    bool isEmpty(){
        return size==0;
    }
};

int main(){
    AVLtree t;
    cout << t.initiation()->data;
    return 0;
}