pat甲级1123Is It a Complete AVL Tree (30分)

pat甲级1123Is It a Complete AVL Tree (30分)

模板题；代码较长，半小时敲完。。。。。。打字太慢

1.AVL插入结点模板

struct node
{
    int data,height=1;
    node *lchild=NULL,*rchild=NULL;
    node(int _data) {data=_data;}
}

int getheight(node *root)
{
    if(root==NULL) return 0;
    else return root->height;
}

void updateheight(node *root)
{
    int l=getheight(root->lchild)+1,r=getheight(root->rchild)+1;
    root->height=r>l?r:l;
}

int getbalancefactor(node *root) 
{
    return getheight(root->lchild)-getheight(root->rchild);
}

//左旋
void L(node *&root)
{
    node *temp=root->rchild;
    root->rchild=temp->lchild;
    temp->lchild=root;
    //顺序不能颠倒，从下往上更新高度
    updateheight(root);
    updateheight(temp);
    root=temp;
}

//右旋
void R(node *&root)
{
    node *temp=root->lchild;
    root->lchild=temp->rchild;
    temp->rchild=root;
    //顺序不能颠倒，从下往上更新高度
    updateheight(root);
    updateheight(temp);
    root=temp;
}

void Insert(node *&root,int v)
{
    if(root==NULL)  root=new node(v);
    else if(root->data > v)
    {
        Insert(root->lchild,v);
        updateheight(root);
        if(getbalancefactor(root)==2)
        {
            if(getbalancefactor(root->lchild)==1)  R(root);
            //else if不是if;
            else if(getbalancefactor(root->lchild)==-1) {L(root->lchild);R(root);}
        }
    }
    else 
    {
        Insert(root->rchild,v);
        updateheight(root);//
        if(getbalancefactor(root)==-2)
        {
            if(getbalancefactor(root->rchild)==-1)  L(root);//
            else if(getbalancefactor(root->rchild)==1) {R(root->rchild);L(root);}
        }
    }
}

2.层序遍历模板

//附带输出
void Level_order(node *root,int n)
{
    queue<node *> q;
    q.push(root);
    int cnt=0;
    bool flag=true,exist=false;//exist=false表示无空子节点
    while(!q.empty())
    {
        node *temp=q.front();
        ++cnt;
        q.pop();
        if(temp->lchild) {if(exist) flag=false;q.push(temp->lchild);}
        else exist=true;
        if(temp->rchild) {if(exist) flag=false;q.push(temp->rchild);}
        else exist=true;
        printf("%d%c",temp->data,cnt<n?' ':'\n');
    }
    printf("%s\n",flag?"YES":"NO");
}

3.主函数

#include <iostream>
#include <queue>
using namespace std;

int main()
{
    int n,temp;
    cin>>n;
    node *Root=NULL;
    for(int i=0;i<n;++i)
    {
        cin>>temp;
        Insert(Root,temp);//建树
    }
    Level_order(Root,n);//层序遍历
    return 0;
}