将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <queue>
using namespace std;
const int N=100;
typedef struct edge* node;
struct edge
{
int data;
node l;
node r;
};
node insert(node BST, int x)
{
if(!BST)
{
BST=(node)malloc(sizeof(struct edge));
BST->data=x;
BST->l=BST->r=NULL;
}
else
{
if(x>=BST->data)
BST->l=insert(BST->l, x);
else
BST->r=insert(BST->r, x);
}
return BST;
}
void traversal1(node BST)
{
queue<node> q;
q.push(BST);
int i=0;
while(q.size())
{
if(q.front()->l) q.push(q.front()->l);
if(q.front()->r) q.push(q.front()->r);
if(!i)
{
printf("%d", q.front()->data);
i++;
}
else
printf(" %d", q.front()->data);
q.pop();
}
}
bool traversal2(node BST)
{
queue<node> q;
q.push(BST);
int flag=0, emp=0;
while(q.size())
{
node f=q.front();
q.pop();
if(f)
{
q.push(f->l);
q.push(f->r);
if(emp)
return false;
}
else
emp=1;
}
return true;
}
int main()
{
int n;
scanf("%d", &n);
node BST=NULL;
for(int i=0;i<n;i++)
{
int x;
scanf("%d", &x);
BST=insert(BST, x);
}
traversal1(BST);
printf("\n");
if(traversal2(BST))
puts("YES");
else
puts("NO");
return 0;
}