[枚举]是否完全二叉搜索树 PTA L3-010

将一系列给定数字顺序插入一个初始为空的二叉搜索树（定义为左子树键值大，右子树键值小），你需要判断最后的树是否一棵完全二叉树，并且给出其层序遍历的结果。

输入格式：

输入第一行给出一个不超过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;
}