PAT 1123. Is It a Complete AVL Tree (30)

本题是属于构造avl树再进行层次遍历。主要难点在于构造avl树，构造过程我在代码当中也有比较详细的注释，希望大家共同交流，指出代码中的一些瑕疵。

1.做这题的思路主要是构造出avl树之后，用一个数组存储树中节点的编号。因为是avl树，所以树不会太高，数组只要开一点就好。我也没去算要开多大，但是开个1,2百肯定是够用的。

2.对于层次遍历，输出的顺序就是在数组中所在位置的先后，只要将数组中元素不等于初始值的元素输出，再对个数进行判断就行。

3.对于完全二叉树，只要判断数组中是否有中间元素之等于初始值就行。

下面是代码：

#include<iostream>
#include<algorithm>
using namespace std;
struct Node
{
    int data;
    Node *left,*right;
};
int getHeight(Node* root)
{
    if(root==NULL)
        return 0;
    return max(getHeight(root->left),getHeight(root->right))+1;
}
Node* leftRotation(Node* root)
{
    //左旋转即交换root与root->right的指针
    Node* temp=root->right;
    root->right=temp->left;//也可以设为NULL
    temp->left=root;
    return temp;
}
Node* rightRotation(Node* root)
{
    Node* temp=root->left;
    root->left=temp->right;
    temp->right=root;
    return temp;
}
Node* insertNode(Node* &root,int data)
{
    if(root==NULL)
    {
        root=new Node();
        root->data=data;
        root->left=root->right=NULL;
    }
    else if(data<root->data)
    {
        root->left=insertNode(root->left,data);//每一步插入都要获得孩子的地址
        if(getHeight(root->left)-getHeight(root->right)==2)//因为插入在左子树，所以只有可能是左子树高于右子树
        {
            if(getHeight(root->left->left)-getHeight(root->left->right)==1)
            {
                //左子树的左子树比右子树高，LL型，向右旋转
                root=rightRotation(root);
            }
            else//LR型，先左旋后右旋
            {
                root->left=leftRotation(root->left);
                root=rightRotation(root);
            }
        }
    }
    else//与插入左子树类似，一步一步进行判断
    {
        root->right=insertNode(root->right,data);
        if(getHeight(root->right)-getHeight(root->left)==2)//因为插入在右子树，所以只有可能是右子树高于左子树
        {
            if(getHeight(root->right->right)-getHeight(root->right->left)==1)
            {
                //左子树的左子树比右子树高，RR型，向右旋转
                root=leftRotation(root);
            }
            else//LR型，先左旋后右旋
            {
                root->right=rightRotation(root->right);
                root=leftRotation(root);
            }
        }
    }
    return root;
}
int level[205];
void levelOrder(Node* root,int index)
{
    if(root==NULL)
        return;
    level[index]=root->data;
    levelOrder(root->left,index*2+1);
    levelOrder(root->right,index*2+2);
}
int main()
{
    fill(level,level+205,-1);
    int n;
    scanf("%d",&n);
    Node *root=NULL;
    for(int i=0;i<n;i++)
    {
        int data;
        scanf("%d",&data);
        root=insertNode(root,data);
    }
    levelOrder(root,0);
    int num=0;//标志当前已经输出的个数，达到n就跳出
    bool flag=true;//标志是否是完全二叉树。因为完全二叉树在数组中是连续存储的
    for(int i=0;i<205;i++)
    {
        if(level[i]!=-1)
        {
            printf("%d",level[i]);
            num++;
            if(num<n)
                printf(" ");
            else
            {
                printf("\n");
                break;
            }
        }
        else
            flag=false;
    }
    if(flag==true)
        printf("YES\n");
    else
        printf("NO\n");
    return 0;
}