5-13 是否完全二叉搜索树 (30分)
将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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<bits/stdc++.h>
using namespace std;
const int maxn = 5e6+5;
int tree[maxn], n;
void add(int r, int val)
{
if(tree[r] == -1)
{
tree[r] = val;
return ;
}
if(val > tree[r]) add(r*2, val);
else add(r*2+1, val);
}
int main(void)
{
while(cin >> n)
{
memset(tree, -1, sizeof(tree));
for(int i = 1; i <= n; i++)
{
int t;
scanf("%d", &t);
add(1, t);
}
bool ok = 1;
int r = 1<<n;
for(int i = 1; i <= r; i++)
{
if(i <= n && tree[i] == -1) ok = 0;
else if(tree[i] != -1)
printf("%s%d", i==1 ? "" : " ", tree[i]);
}
printf("\n%s\n", ok ? "YES" : "NO");
}
return 0;
}