PAT - 天梯赛 L3-010 是否完全二叉树（dfs）

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

思路：因为这个题的根节点就是给出的第一个数，左右子节点都是遇到的第一个比自己大（小）的数，所以用一个dfs 从第一个开始从上往下遍历，若比此数大则为左子树 否则右子树。

#include<cstdio>
#include<cstring>
using namespace std;

int n, arr[42]; //  注意数组长度，有两个wa就是长度不够 
void build(int l, int a){
	if(arr[l] == 0){
		arr[l] = a;
	}
	else if(arr[l] < a) build(2*l, a);
	else build(2*l+1, a);
}
int main(){	
	scanf("%d", &n);
	memset(arr, 0, sizeof(arr));
	for(int i = 0; i < n; i++){
		int a;
		scanf("%d", &a);
		build(1, a);
	}
	int flag = 0, cnt = 0, i = 1;
	while(cnt < n){
		if(arr[i] == 0) flag = 1;
		else{
			printf("%d", arr[i]);
			if(cnt != n-1) printf(" ");
			cnt++;
		}
		i++;
	}
	if(flag) printf("\nNO");
	else printf("\nYES");
	return 0;
}