L3-010. 是否完全二叉搜索树(数组模拟树)

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

// 这题就是用数组模拟二叉树，然后按题意插入数据，记得数组开大一点，树可能会有20层
#include <iostream>
#define max_size 1048576
using namespace std;
int tree[max_size + 1] = { 0 };

void insert(int root, int num) {
	if (tree[root] == 0) {
		tree[root] = num;
		return;
	}
	else if (num > tree[root]) {
		if (tree[root * 2] == 0) {
			tree[root * 2] = num;
			return;
		}
		else
			insert(root * 2, num);
	}
	else {
		if (tree[root * 2 + 1] == 0) {
			tree[root * 2 + 1] = num;
			return;
		}
		else
			insert(root * 2 + 1, num);
	}
	return;
}

int main() {
	//freopen("1.txt", "r", stdin);
	int N;
	cin >> N;
	for (int i = 0; i < N; i++) {
		int num;
		cin >> num;
		insert(1, num);
	}

	bool flag = true;
	int num_node = 1;
	for (int i = 1; i <= max_size; i++) {
		if (num_node > N)
			break;
		if (tree[i] != 0) {
			if (i != 1)
				cout << " ";
			cout << tree[i];
			num_node++;
		}
		else
			flag = false;
	}
	cout << endl;
	if (flag)
		cout << "YES";
	else
		cout << "NO";
	return 0;
}