问题描述:
将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过20的正整数N;第二行给出N个互不相同的正整数,其间以空格分隔。
输出格式:
将输入的N个正整数顺序插入一个初始为空的二叉搜索树。在第一行中输出结果树的层序遍历结果,数字间以1个空格分隔,行的首尾不得有多余空格。第二行输出YES,如果该树是完全二叉树;否则输出NO。
解法:
数组模拟二叉树,节点x的左子节点是2x,右子节点是2x+1。
判断是否是完全二叉树只需要判断是否最大下标==n即可(即只有[1,n]被填满)
code:
#include<bits/stdc++.h>
using namespace std;
const int maxm=1e5+5;
int a[maxm<<2];
int ma;
int n;
void add(int val,int node){
if(!a[node]){
a[node]=val;
ma=max(ma,node);
return ;
}
if(val>a[node]){//左大
add(val,node*2);
}else{//右小
add(val,node*2+1);
}
}
signed main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
int x;scanf("%d",&x);
add(x,1);
}
for(int i=1;i<=ma;i++){
if(a[i]){
printf("%d%c",a[i],i==ma?'\n':' ');
}
}
if(ma==n)puts("YES");
else puts("NO");
return 0;
}