17、樹的子結構
題目描述:
輸入兩棵二叉樹A,B,判斷B是不是A的子結構。(ps:我們約定空樹不是任意一個樹的子結構)
思路:
第一部分先判斷是否有相等的根節點。把root2根節點分別和root1的根節點、左孩子、右孩子逐一比較,因需要多次比較才能確定故設置標誌位記錄比較結果。
第二部分是已經找到相等的根節點之後以該根節點爲起點判斷是否包含整棵樹。判斷時既要判斷左孩子又要判斷右孩子,此過程是遞歸的。過程中有任何一個節點值不等那麼則返回false;如果root1比較完了,root2還沒比較完也返回false。當root2比較完時返回true。
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public 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)||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);
}
}
**注:**在第一個比較根節點的函數中,不直接使用return而設置result記錄結果是因爲存在一種情況,即根節點相同但是比較左右孩子過程中發現不相同,這時如果使用return則直接返回false了,但是有可能在這個根節點的左右子樹中存在和root2完全相同的子樹,直接return會漏掉這個情況,直接return並沒有完成對整棵樹的搜索。