04-樹4 是否同一棵二叉搜索樹(25 分)【java語言實現】
給定一個插入序列就可以唯一確定一棵二叉搜索樹。然而,一棵給定的二叉搜索樹卻可以由多種不同的插入序列得到。例如分別按照序列{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
思路
按照 小白專場中的代碼用java 重寫即可,需要注意一些小細節。
代碼
import java.util.Scanner;
class BinarySearchTree{
int value;
BinarySearchTree left;
BinarySearchTree right;
int flag;
/**
* 夠着函數,用於初始化
*
* https://blog.csdn.net/zjkC050818/article/details/53730108
* @param value
*/
public BinarySearchTree(int value){
this.value = value;
left = null;
right = null;
flag =0;
}
}
public class Main {
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");
}
// sc.nextInt();
length=sc.nextInt();
}
}
/**
* 檢測函數,這個不懂
* @param value
* @param t
* @return
*/
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;
}
}
/**
* 建立二叉搜索樹
* @param sc
* @param length
* @return
*/
public static BinarySearchTree buildTree(Scanner sc,int length){
BinarySearchTree root = null;
int tem;
for (int i=0;i<length;i++){
tem = sc.nextInt();//讀入
root = insert(tem,root);
}
return root;//返回根結點
}
/**
* 插入結點
* @param value
* @param t
* @return
*/
public static BinarySearchTree insert(int value,BinarySearchTree t){
if (t == null){/**如果爲null,則實例化結點*/
return new BinarySearchTree(value);
}
else {
if (value>t.value){
t.right = insert(value,t.right);
}
else if (value<t.value){
t.left = insert(value,t.left);
}
/**如果已經存在,則什麼都不做*/
}
return t;
}
/**
* 判斷是否合格,一個一個的 check
* @param root
* @param sc
* @param length
* @return
*/
public static boolean judge(BinarySearchTree root,Scanner sc,int length) {
int temp;
boolean flag = true;/**false 表示還是一致的*/
for (int j = 0; j < length; j++) {
temp = sc.nextInt();
if (!check(temp, root)) {
//需要繼續打印
j++;//因爲這裏在for 循環本應該加1 的,但是被耽誤了,不加一就會出現錯位
while (j < length) {
int x= sc.nextInt();
j++;
}
return false;
}
}
return true;
}
/**
* 重置 二叉樹中的 flag
* @param root
*/
public static void reset(BinarySearchTree root){
if (root.left != null)
reset(root.left);
if (root.right != null)
reset(root.right);
root.flag =0;
}
}
參考博客:zjkC050818的博客