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

#include <stdio.h>
#include <iostream>
#include <vector>

using namespace std;

vector<int> tree(20*2+10, -1);

void add(int num, int index)
{
	if(tree[index] == -1)
	{
		tree[index] = num;
	}
	else if(num >= tree[index])
	{
		add(num, 2*index);
	}
	else
	{
		add(num, 2*index+1);
	}
}

int main()
{
	int n;
	
	cin >> n;
	for(int i = 0; i < n; ++i)
	{
		int m;
		cin >> m;
		add(m, 1);
	}
	
	bool flag = 0;
	for(int i = 1; i <= n; ++i)
	{
		if(tree[i] == -1)
		{
			flag = 1;
			break;
		}
	}
	
	if(!flag)
	{
		for(int i = 1; i <= n; ++i)
		{
			if(i != 1) cout << " ";
			cout << tree[i];
		}
		cout << endl << "YES";
	}
	else
	{
		int cnt = 0;
		for(int i = 1; i < 2*n+10; ++i)
		{
			if(tree[i] != -1 && cnt != (n-1))
			{
				cout << tree[i] << " ";
				++cnt;
			}
			else if(tree[i] != -1)
			{
				cout << tree[i] << endl << "NO";
			}
		}
	}
	
	return 0;
}