题目:
将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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;
int s[500];
int ma=1;//一定设置为1,有可能n为1进不了if(s[id*2])||if(s[id*2+1])
int main(){
int n;
cin>>n;
for(int i=1;i<=n;i++){
int x,id=1;
cin>>x;
while(s[id]){
if(s[id]>x){
id=id*2+1;
}else{
id=id*2;
}
}
s[id]=x;
}
queue<int>q;
q.push(1);
while(!q.empty()){
int id=q.front();
q.pop();
if(id!=1){
cout<<" ";
}
cout<<s[id];
if(s[id*2]){
q.push(id*2);
ma=max(ma,id*2);
}
if(s[id*2+1]){
q.push(id*2+1);
ma=max(ma,id*2+1);
}
}
cout<<endl;
if(ma==n){
cout<<"YES";
}else{
cout<<"NO";
}
return 0;
}