将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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 <iostream>
using namespace std;
const int MAX = 20 + 10;
int tem, last; //临时变量,结点最大编号
int tree[MAX]; //二叉搜索树(0为空)
void creatTree(int x) //建立二叉搜索树
{
if (tree[x] == 0)
{
tree[x] = tem;
last = max(last, x);
}
else
{
if (tem > tree[x])
creatTree(x * 2);
else
creatTree(x * 2 + 1);
}
return;
}
int main()
{
int n; cin >> n;
for (int i = 0; i < n; i++)
{
cin >> tem;
creatTree(1);
}
bool flag = true; //是否是完全二叉树
for (int i = 1; i <= last; i++)
{
if (tree[i] && i != last)
cout << tree[i] << " ";
else if (tree[i])
cout << tree[i] << endl;
else
flag = false;
}
if (flag)
cout << "YES" << endl;
else
cout << "NO" << endl;
return 0;
}
注意事项:
建立二叉搜索树的同时记录层序遍历序列,详见代码。
如有问题,欢迎提出。