題目描述
輸入兩棵二叉樹A,B,判斷B是不是A的子結構。(ps:我們約定空樹不是任意一個樹的子結構)
題目分析
第一種方法:判斷B樹是不是A樹的子結構,我們可以找出A中所有與B數根節點相等的元素,再以該元素爲根節點遍歷,判斷以該節點爲根節點的樹是否包含B樹的所有節點。
第二種方法:可以創建兩個列表(ArrayList),分別儲存兩個樹的所有值(通過前序遍歷得到),然後使用ContainAll方法判斷列表A是否包含列表B。如果包含說明B樹是A樹的子結構。否則則反之。利用空間換時間。
Java實現
- 第一種方法:
public static boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean result = false;
//當Tree1和Tree2都不爲零的時候,才進行比較。否則直接返回false
if (root2 != null && root1 != null) {
//如果找到了對應Tree2的根節點的點
if(root1.val == root2.val){
//以這個根節點爲爲起點判斷是否包含Tree2
result = doesTree1HaveTree2(root1,root2);
}
//如果找不到,那麼就再去root的左兒子當作起點,去判斷時候包含Tree2
if (!result) {
result = HasSubtree(root1.left,root2);
}
//如果還找不到,那麼就再去root的右兒子當作起點,去判斷時候包含Tree2
if (!result) {
result = HasSubtree(root1.right,root2);
}
}
//返回結果
return result;
}
public static boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
//如果Tree2已經遍歷完了都能對應的上,返回true
if (node2 == null) {
return true;
}
//如果Tree2還沒有遍歷完,Tree1卻遍歷完了。返回false
if (node1 == null) {
return false;
}
//如果其中有一個點沒有對應上,返回false
if (node1.val != node2.val) {
return false;
}
//如果根節點對應的上,那麼就分別去子節點裏面匹配
return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
}
- 第二種方法:
import java.util.ArrayList;
public class Solution {
public ArrayList<Integer> list1 = new ArrayList<Integer>();//用於儲存樹A的節點值
public ArrayList<Integer> list2 = new ArrayList<Integer>();//用於儲存樹B的節點值
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
//當Tree1和Tree2都不爲零的時候,才進行遍歷。否則直接返回false
if(root1 == null || root2 == null){
return false;
}
traverse1(root1);
traverse2(root2);
return list1.containsAll(list2);//判斷列表A是否包含列表B
}
//通過遞歸前序遍歷樹A
public void traverse1(TreeNode root){
if(root != null){
list1.add(root.val);
traverse1(root.left);
traverse1(root.right);
}
}
//通過遞歸前序遍歷樹B
public void traverse2(TreeNode root){
if(root != null){
list2.add(root.val);
traverse2(root.left);
traverse2(root.right);
}
}
}