将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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>
#include<queue>
using namespace std;
struct bis {
int data;
bis* left;
bis* right;
};
bis* insert(bis* t, int k) {
bis* pre=NULL, *now;
now = t;
while (now) {
if (k == now->data) {
return t;
}
else {
pre = now;
if (k > now->data) {
now = now->left;
}
else {
now = now->right;
}
}
}
bis* newnode = (bis*)malloc(sizeof(bis));
newnode->data = k;
newnode->left = NULL;
newnode->right = NULL;
if (!t) {
t = newnode;
}
else {
if (k > pre->data) {
pre->left = newnode;
}
else {
pre->right = newnode;
}
}
return t;
}
bis* creat() {
bis* root = NULL;
int n;
cin >> n;
int temp;
for (int i = 0; i < n; i++) {
cin >> temp;
root = insert(root, temp);
}
return root;
}
void leveltraverse(bis* t) {
if (t == NULL) {
return;
}
queue<bis*> q;
cout << t->data;
q.push(t);
bis* temp;
while (!q.empty()) {
temp = q.front();
q.pop();
if (temp->left != NULL) {
cout << " " << temp->left->data;
q.push(temp->left);
}
if (temp->right != NULL) {
cout << " " << temp->right->data;
q.push(temp->right);
}
}
}
bool judge(bis* t) {
queue<bis*> q;
q.push(t);
int flag = 0;
bis* temp;
if (!t) {
return true;
}
while (!q.empty()) {
temp = q.front();
q.pop();
if (!temp) {
flag = 1;
}
else {
if (flag == 1) {
return false;
}
q.push(temp->left);
q.push(temp->right);
}
}
return true;
}
int main()
{
bis* root = creat();
leveltraverse(root);
cout << endl;
if (judge(root)) {
cout << "YES" << endl;
}
else {
cout << "NO" << endl;
}
return 0;
}