将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
根据节点的编号判断是否是完全二叉搜索树,如果最大的节点编号不等于n就不是,否则就是。
#include<bits/stdc++.h>
using namespace std;
struct node {
struct node *l,*r;
int val;
};
int n,m,num=0;
struct node *build(struct node *root,int w,int in) {
if(root==NULL) {
root = (struct node *)malloc(sizeof(struct node));
root->val = w;
root->l = NULL;
root->r = NULL;
num = max(num,in);
return root;
}
if(root->val > w) root->r=build(root->r,w,in*2+1);
else root->l = build(root->l,w,in*2);
return root;
}
void bfs(struct node *root) {
queue<node*>q;
q.push(root);
cout<<root->val;
while(!q.empty()) {
struct node *top;
top=q.front();
q.pop();
if(top->l) {
cout<<' '<<top->l->val;
q.push(top->l);
}
if(top->r) {
cout<<' '<<top->r->val;
q.push(top->r);
}
}
}
int main() {
cin>>n;
struct node *root=NULL;
for(int i=1; i<=n; i++) {
int k;
cin>>k;
root = build(root,k,1);
}
bfs(root);
cout<<endl;
if(num==n) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
return 0;
}