PTA 7-25 是否完全二叉搜索树 (30 分)

PTA 7-25 是否完全二叉搜索树 (30 分)

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：

输入第一行给出一个不超过20的正整数N；第二行给出N个互不相同的正整数，其间以空格分隔。

输出格式：

将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果，数字间以1个空格分隔，行的首尾不得有多余空格。第二行输出YES，如果该树是完全二叉树；否则输出NO。

输入样例1：

9
38 45 42 24 58 30 67 12 51

输出样例1：

38 45 24 58 42 30 12 67 51
YES

输入样例2：

8
38 24 12 45 58 67 42 51

输出样例2：

38 45 24 58 42 12 67 51
NO

思路：先储存，再遍历满二叉树。（当时把满二叉树和完全二叉树记岔劈了，哎。。。）

代码：

#include<bits/stdc++.h>
using namespace std;
int n,x;
int a[1048580]={0};//2的20次方最大1048576
int flag=1;//做标记
void bfs()
{
    for(int i=0;i<n;i++)
    {
        cin>>x;
        int t=1;
        while(t)
        {
            if(a[t]==0)
            {
                a[t]=x;
                break;
            }
            else if(a[t]<x)
                t=t*2;
            else
                t=t*2+1;
        }
    }
}
void check()
{
    int cnt=0;
    for(int i=1;i<=1048576;i++)//遍历完全二叉树
    {
        if(a[i]!=0)
        {
            cnt++;
            cout << a[i];
            cout << (cnt == n ? "\n" : " ");
            if(cnt==n)
                break;
        }
        else
            flag=0;
    }
}
int main()
{
    cin>>n;
    bfs();
    check();
    if(flag)
        cout<<"YES";
    else
        cout<<"NO";
}