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");

}