将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
题意: 给出一个序列,先构造出一棵二叉搜索树,之后输出其层次遍历,最后判断是否是一棵完全二叉树。
分析: 这道题如果建图跑bfs就太复杂了,实际上点的个数非常少,最多才20个,完全可以用数组存下这棵二叉树,而这样存的好处在于判断是否为完全二叉树会非常方便,只需要遍历一下1~n,如果都填满了数据那就说明是个完全二叉树,而输出层次遍历也很方便,直接遍历1~2e6,只要有数据就输出。
具体代码如下:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <string>
#include <vector>
using namespace std;
int w[105], tr[2000005];
void build(int now, int x){
if(tr[now] == -1){
tr[now] = x;
return;
}
if(x < tr[now]) build(now<<1|1, x);
else build(now<<1, x);
}
signed main()
{
int n;
cin >> n;
memset(tr, -1, sizeof tr);
for(int i = 1; i <= n; i++){
cin >> w[i];
build(1, w[i]);
}
bool flag = false;
for(int i = 1; i <= n; i++)
if(tr[i] == -1){
flag = true;
break;
}
int cnt = 0;
for(int i = 1; i <= 2000000; i++){
if(tr[i] != -1){
if(cnt == 0) printf("%d", tr[i]);
else printf(" %d", tr[i]);
cnt++;
}
if(cnt == n) break;
}
puts("");
if(flag) puts("NO");
else puts("YES");
return 0;
}