将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
主要是完全二叉树的性质,不难,感觉就这道题做的最顺利了QAQ
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
using namespace std;
struct node{
int data, id;
struct node *lchild, *rchild;
node(int x): data(x), id(0), lchild(0), rchild(0){}
};
void Insert(node *&root, int x)
{
if(!root)
{
root = new node(x);
return;
}
if(x > root->data)
{
Insert(root->lchild, x);
}
else
{
Insert(root->rchild, x);
}
}
void Layer(node *root)
{
queue<node*> q;
bool flag = true, ans = true;
int k = 0;
root->id = 1;
q.push(root);
while(!q.empty())
{
node *temp = q.front();
q.pop();
k++;
if(temp->id != k)
ans = false;
if(flag)
{
printf("%d", temp->data);
flag = false;
}
else
printf(" %d", temp->data);
if(temp->lchild)
{
temp->lchild->id = 2 * temp->id;
q.push(temp->lchild);
}
if(temp->rchild)
{
temp->rchild->id = 2 * temp->id + 1;
q.push(temp->rchild);
}
}
if(ans)
printf("\nYES");
else
printf("\nNO");
}
int main()
{
int n, m;
scanf("%d", &n);
node *root = NULL;
while(n--)
{
scanf("%d", &m);
Insert(root, m);
}
Layer(root);
return 0;
}