5-2 是否完全二叉搜索树 (30分)
将一系列给定数字顺序插入一个初始为空的二叉搜索树(定义为左子树键值大,右子树键值小),你需要判断最后的树是否一棵完全二叉树,并且给出其层序遍历的结果。
输入格式:
输入第一行给出一个不超过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
package BinaryTree;
import java.util.Scanner;
public class BinarySearchTree {
//9
//38 45 42 24 58 30 67 12 51
/*
* 用数组进行存储二叉树,根据前n个数是否值为0判断是否为完全二叉树
*/
static int[] a=new int[2200000];
public static void main(String[] args) {
BinarySearchTree m=new BinarySearchTree();
Scanner in=new Scanner(System.in);
int n=in.nextInt();
for(int i=0;i<n;i++){
int data=in.nextInt();
m.insert(1,data);
}
boolean flag=true;
for(int i=1;i<=n;i++){
if(a[i]==0){//判断是否为完全二叉树
flag=false;
break;
}
}
int r=1;
int i=1;
while(r<=n){//用于输出
if(a[i]!=0){
System.out.print(a[i]);
r++;
if(r<=n){
System.out.print(" ");
}
}
i++;
}
System.out.println();
System.out.println(flag==true?"YES":"NO");
}
private void insert(int i, int data) {
if(a[i]==0){
a[i]=data;
}else{
if(data>a[i]){
insert(2*i,data);
}else{
insert(2*i+1,data);
}
}
}
}