您的位置:首页 > 编程语言 > Java开发

PAT:是否为同一棵二叉搜索树(Java实现)

2016-12-18 21:30 441 查看
是否同一棵二叉搜索树   (25分)

给定一个插入序列就可以唯一确定一棵二叉搜索树。然而,一棵给定的二叉搜索树却可以由多种不同的插入序列得到。例如分别按照序列{2, 1, 3}和{2, 3, 1}插入初始为空的二叉搜索树,都得到一样的结果。于是对于输入的各种插入序列,你需要判断它们是否能生成一样的二叉搜索树。

输入格式:

输入包含若干组测试数据。每组数据的第1行给出两个正整数N (≤10)和L,分别是每个序列插入元素的个数和需要检查的序列个数。第2行给出N个以空格分隔的正整数,作为初始插入序列。最后L行,每行给出N个插入的元素,属于L个需要检查的序列。

简单起见,我们保证每个插入序列都是1到N的一个排列。当读到N为0时,标志输入结束,这组数据不要处理。

输出格式:

对每一组需要检查的序列,如果其生成的二叉搜索树跟对应的初始序列生成的一样,输出“Yes”,否则输出“No”。

输入样例:

4 2
3 1 4 2
3 4 1 2
3 2 4 1
2 1
2 1
1 2
0

输出样例:

Yes
No
No


注意:程序输出并不完善,只能同时输出一组判定结果。

import java.util.Scanner;

class BinarySearchTree{
int value;
BinarySearchTree left;
BinarySearchTree right;
int flag;
public BinarySearchTree(int value){
this.value=value;
left=null;
right=null;
flag=0;
}
}

public class JudgeTree {

public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner sc=new Scanner(System.in);
int length=sc.nextInt();
int N=0;
BinarySearchTree root=null;
while(length!=0){
N=sc.nextInt();
root=buildTree(sc,length);
//判断是否同一棵树
boolean[] arr=new boolean
;
for(int i=0;i<N;i++){
arr[i]=judge(root,sc,length);
reset(root);
}

for(int j=0;j<N;j++){
if(arr[j])
System.out.println("YES");
else
System.out.println("NO");
}
length=sc.nextInt();
}

}

public static boolean check(int value,BinarySearchTree t){
if(t.flag==0){
if(value==t.value){
t.flag=1;
return true;
}
else
return false;
}
if(value>t.value){
return check(value,t.right);
}
if(value<t.value){
return check(value,t.left);
}else{
t.flag=1;
return true;
}
}

public static BinarySearchTree buildTree(Scanner sc,int length){
BinarySearchTree root=null;
int tmp;
for(int i=0;i<length;i++){
tmp=sc.nextInt();
root=insert(tmp,root);
}
return root;
}
public static BinarySearchTree insert(int value,BinarySearchTree t){
if(t==null){
return new BinarySearchTree(value);
}
if(value>t.value){
t.right=insert(value,t.right);
}else{
t.left=insert(value,t.left);
}
return t;
}

public static boolean judge(BinarySearchTree root,Scanner sc,int length){
int tmp;
for(int j=0;j<length;j++){
tmp=sc.nextInt();
if(!check(tmp,root)){
while(j<length){
sc.nextInt();
j++;
}
return false;
}
}
return true;
}
public static void reset(BinarySearchTree root){
if(root!=null){
root.flag=0;
reset(root.left);
reset(root.right);
}
}
}

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: