劍指Offer--樹的子結構
阿新 • • 發佈:2018-08-16
left tree res 返回 als pub 代碼 有一個 沒有
問題描述:輸入兩棵二叉樹A,B,判斷B是不是A的子結構。(ps:我們約定空樹不是任意一個樹的子結構)
思路:先在樹A中找到和樹B的根節點的值一樣的結點R;然後判斷樹A中以R為根節點的子樹是不是包含和樹B一樣的結構。
代碼:
/** public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } */ public class Solution {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){ //以這個根節點為為起點判斷是否包含Tree2result = 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); } }
劍指Offer--樹的子結構