题意
将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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>
using namespace std;
int tree[1<<21];
void build_tree(int pos,int vnode)
{
if(tree[pos]==0)
tree[pos]=vnode;
else
{
if(tree[pos]<vnode)
build_tree(2*pos+1,vnode);
else
build_tree(2*pos+2,vnode);
}
}
int main()
{
int N,x;
bool flag=true;
scanf("%d",&N);
for(int i=0;i<N;i++)
{
scanf("%d",&x);
build_tree(0,x);
}
int countt=0;
for(int i=0;countt<N;i++)
{
if(!tree[i])
flag=false;
else
{
printf("%d",tree[i]);
if(countt!=N-1)
printf(" ");
countt++;
}
}
printf("\n");
if(flag)
printf("YES\n");
else
printf("NO\n");
return 0;
}