1123.（重、错）Is It a Complete AVL Tree

题意：给定结点个数n和插入序列，判断构造的AVL树是否是完全二叉树？

思路：AVL树的建立很简单。而如何判断是不是完全二叉树呢？通过层序遍历进行判断：当一个结点的孩子结点为空时，则此后就不能有新的结点入队。若没有，则是完全二叉树，否则不是。

代码：

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
 
vector<int> layer;
 
struct Node {
    int v, height;
    Node *lchild, *rchild;
};
 
Node* newNode(int v) {
    Node* pNode = new Node;
    pNode->v = v;
    pNode->height = ;
    pNode->lchild = pNode->rchild = NULL;
    return pNode;
}
 
int getHeight(Node* root){
 
    if(root==NULL) return ;
    return root->height;
}
void updateHeight(Node* root) {
    root->height = max(getHeight(root->lchild), getHeight(root->rchild))+;
}
 
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 = newNode(v);
        return;
    }
 
    if (v < root->v) {
        insert(root->lchild,v);
        updateHeight(root);
        if (getBalanceFactor(root) == ) {
                if(getBalanceFactor(root->lchild)==){
                    R(root);
                }else if(getBalanceFactor(root->lchild)==-){
                    L(root->lchild);
                    R(root);
                }
 
        }
    }
    else {
        insert(root->rchild,v);
        updateHeight(root);
        if (getBalanceFactor(root) == -) {
            if(getBalanceFactor(root->rchild)==-){
                L(root);
            }
            else if(getBalanceFactor(root->rchild)==){
                R(root->rchild);
                L(root);
            }
        }
    }
}
bool isComplete =true;
int after=;
void layerOrder(Node* root){
    queue<Node*> Q;
    Q.push(root);
    while(!Q.empty()){
        Node* front=Q.front();
        Q.pop();
        layer.push_back(front->v);
 
        if(front->lchild!=NULL){
            if(after==) isComplete=false;
            Q.push(front->lchild);
        }else{
            after=;
        }
 
        if(front->rchild!=NULL){
            if(after==) isComplete=false;
            Q.push(front->rchild);
        }else{
            after=;
        }
    }
 
}
 
//vector<int> insertOrder;
 
int main()
{
    int n,data;
    scanf("%d",&n);
    Node* root=NULL;
    for(int i=;i<n;i++){
        scanf("%d",&data);
        insert(root,data);
    }
    layerOrder(root);
 
    for(int i=;i<layer.size()-;i++){
        printf("%d ",layer[i]);
    }
    printf("%d\n",layer[n-]);
    printf("%s\n",isComplete==true?"YES":"NO");
 
    return ;
}