将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
题解
主要难在判二叉树是否完全
本来是想用数组模拟,只要判每次有没有超就可以了,但是如果每次都是右节点的话要2^20≈1e6的数组,而且层序遍历时候要一个一个遍历判
所以用结构体(只要20个),先建树,之后bfs时候给每个节点标序号,如果序号超过n就否
#include <bits/stdc++.h>
using namespace std;
struct Node{
int data,id;
Node *l=NULL,*r=NULL;
};
Node *head;
int n,num,flag;
int tt;
void creat(Node *&root,int num){
if(root==NULL){
root=new Node;
root->data=num;
}else{
if(root->data<num) creat(root->l,num);
else creat(root->r,num);
}
}
queue<Node*> q;
void bfs(Node *root){
root->id=1;
q.push(root);
while(q.size()){
Node *f=q.front();
q.pop();
if(f!=root) cout<<" ";
cout<<f->data;
if(f->id>n) flag=0;
if(f->l!=NULL){
q.push(f->l);
f->l->id=f->id<<1;
}
if(f->r!=NULL){
q.push(f->r);
f->r->id=(f->id<<1)|1;
}
}
}
int main(int argc, char** argv) {
cin>>n;
for(int i=0;i<n;i++){
cin>>num;
creat(head,num);
}
flag=1;
bfs(head);
puts("");
if(flag) puts("YES");
else puts("NO");
return 0;
} 
京公网安备 11010502036488号