将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
主要问题就是构造这个树,可以看出第一个元素是根节点,其后大于a[1]和小于a[1]的元素分别构成两个左右序列,递归构造即可
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<queue>
using namespace std;
typedef struct bitree{
int data;
struct bitree *lc;
struct bitree *rc;
}bitree;
bool f=1,g=1;
bitree *create(int *a,int n)
{
if(n<=0)return NULL;
bitree *t=(bitree*)malloc(sizeof(bitree));
t->data=a[1];
int left[n+1],right[n+1],l=0,r=0;
for(int i=2;i<=n;i++)
{
if(a[1]>a[i])right[++r]=a[i]; //左右两子树的序列
else left[++l]=a[i];
}
t->lc=create(left,l);
t->rc=create(right,r);
return t;
}
void levelorder(bitree *BT) //层序遍历
{
queue<bitree*> q;
bitree *T;
if(!BT)return ;
q.push(BT);
int i=1;
while(!q.empty()){
T=q.front();
q.pop();
if(i==1)
{
cout<<T->data;
i++;
}
else{
cout<<" "<<T->data;
}
if(!f&&(T->lc||T->rc))g=0; //关于yes,no的判断
if(!T->lc||!T->rc)f=0;
if(!T->lc&&T->rc)g=0;
if(T->lc)
q.push(T->lc);
if(T->rc)
q.push(T->rc);
}
}
int main(){
bitree *T;
int n;
cin>>n;
int a[n+1];
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
T=create(a,n);
levelorder(T);
cout<<endl;
if(g)
cout<<"YES";
else cout<<"NO";
return 0;
}