題目
輸入兩棵二叉樹A,B,判斷B是不是A的子結構。(ps:我們約定空樹不是任意一個樹的子結構)
1.思路
最開始用自己的思路,沒有做對。我的思路是分別對兩個樹進行先/中/後序遍歷,並將結點的值追加到字符串裏,最後判斷字符串1是否包含字符串2,包含就是子樹,反之則不是。
後來想了想,可能會出現並不是子樹,但是先序遍歷後其字符串順序的確是會出現子串關係的情況。還是太菜了。。。
劍指offer的思路:
1. 首先設置標誌位result = false,因爲一旦匹配成功result就設爲true,
剩下的代碼不會執行,如果匹配不成功,默認返回false;
2. 遞歸思想,如果根節點相同則遞歸調用DoesTree1HaveTree2(),
如果根節點不相同,則判斷tree1的左子樹和tree2是否相同,再判斷右子樹和tree2是否相同;
3. HasSubTree中,若兩棵樹都不爲空才進行判斷,DoesTree1HasTree2中,若Tree2爲空,則說明第二棵樹遍歷完了,即匹配成功,tree1爲空有兩種情況
(1)如果tree1爲空&&tree2不爲空說明不匹配,
(2)如果tree1爲空,tree2爲空,說明匹配。
2. 代碼(Java實現)
// 劍指offer的
public class Solution{
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean result = false;
if(root1 != null && root2 != null){
if(root1.val == root2.val){
result = DoesTree1HaveTree2(root1,root2);
}
if(!result){
result = HasSubtree(root1.left, root2);
}
if(!result){
result = HasSubtree(root1.right, root2);
}
}
return result;
}
public boolean DoesTree1HaveTree2(TreeNode root1,TreeNode root2){
if(root2 == null)
return true;
if(root1 == null)
return false;
if(root1.val != root2.val)
return false;
return DoesTree1HaveTree2(root1.left, root2.left) && DoesTree1HaveTree2(root1.right, root2.right);
}
}