PAT：是否为同一棵二叉搜索树（Java实现）

是否同一棵二叉搜索树   (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[N];
			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);
		}
	}
}