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

#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstring>
using namespace std;
int tree[200],le[200],ri[200],cnt=0,path[200],p=1,idd[200];
bool flag=1;
//tree[X] 第X棵树的值 输入的顺序
//le[X] 第X棵树的左子树
//ri[X]同理
//idd[X]保存了第X课树的真正编号。
void insert(int x,int v,int id)
{
    if(tree[x]==0)
    {
        tree[x]=v;
        idd[x]=id;
        cnt++;
    }
    else
    {
        if(v>tree[x])
        {
            if(le[x]==0)
            {
                le[x]=cnt+1;
            }
            insert(le[x],v,id*2);
        }
        else
        {
            if(ri[x]==0)
            {
                ri[x]=cnt+1;
            }
            insert(ri[x],v,id*2+1);
        }
    }
}
void solve()
{
    queue<int> q;
    q.push(0);
    while(!q.empty())
    {
        int c=q.front();
        q.pop();
        if(p<idd[c])//id
        {
            flag=0;
        }
        path[p++]=tree[c];
        if(le[c]&&ri[c])
        {
            q.push(le[c]);
            q.push(ri[c]);
        }
        else
        {
            if(le[c])
                q.push(le[c]);
            if(ri[c])
                q.push(ri[c]);
        }

    }

}
int main()
{
    int n;
    cin>>n;
    for(int i=1; i<=n; i++)
    {
        int tem;
        cin>>tem;
        insert(0,tem,1);
    }
    solve();
    for(int i=1; i<p; i++)
    {
        if(i-1) printf(" ");
        printf("%d",path[i]);
    }
    if(flag) printf("\nYES\n");
    else printf("\nNO\n");

}