天梯赛 L3-010. 是否完全二叉搜索树

L3-010. 是否完全二叉搜索树

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出一个不超过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

数组建树，遍历途中存在空节点，则结果为NO，否则为YES。自己写的有点复杂，参考了大神的博客，发现原来可以写得这么简洁。

#include<iostream>
#include<cstring>
#include<stdio.h>
#include<string>
#include<math.h>
#include<algorithm>

using namespace std;

int tree[1<<20];
int num,n;

void built(int pos)
{
    if(tree[pos]==0)
        tree[pos]=num;
    else
        if(tree[pos] < num)
            built(pos<<1);
        else
            built(pos<<1|1);
}
int main()
{
    cin>>n;
    for(int i = 1; i <= n; i++)
    {
        cin>>num;
        built(1);
    }
    
    bool flag = false;
    for(int i = 1, con = 1; con <= n; i++)
    {
        if(tree[i]==0)
            flag = true;
        else
        {
            cout<<tree[i];
            if(con++!=n)
                cout<<' ';
        }
    }
    cout<<endl;
    if(flag)
        cout<<"NO"<<endl;
    else
        cout<<"YES"<<endl;
    return 0;
}