題目描述

輸入兩棵二叉樹A，B，判斷B是不是A的子結構。（ps：我們約定空樹不是任意一個樹的子結構）

/**
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) {
        
    }
}
 

解題思路

• 首先設定標誌位result = false，因為一旦匹配成功result就設為true，剩下的程式碼不會執行，如果匹配不成功，預設返回false
• 遞迴思想，如果根節點相同則遞迴呼叫DoesTree1HaveTree2（），
• 如果根節點不相同，則判斷tree1的左子樹和tree2是否相同，再判斷右子樹和tree2是否相同
• 注意null的條件，HasSubTree中，如果兩棵樹都不為空才進行判斷，DoesTree1HasTree2中，如果Tree2為空，則說明第二棵樹遍歷完了，即匹配成功，
• tree1為空有兩種情況
  - 如果tree1為空&&tree2不為空說明不匹配，
  - 如果tree1為空，tree2為空，說明匹配。
  */

public class Solution {

    public boolean HasSubtree(TreeNode root1, TreeNode root2) {
        boolean result = false;
        if (root1 == null || root2 == null) {
            return false;
        }
        if (root1 != null && root2 != null) {
            if (root1.val == root2.val) {
                result = doesTree1HaveTree2(root1, root2);
            }
            //如果找不到，把root的左兒子當作起點，去判斷是否包含Tree2
            if (!result) {
                result = HasSubtree(root1.left,root2);
            }
            //如果找不到，用root的右兒子當作起點，去判斷是否包含Tree2
            if (!result) {
                result = HasSubtree(root1.right,root2);
            }
        }
        return result;
    }

    private boolean doesTree1HaveTree2(TreeNode root1, TreeNode root2) {
        //如果Tree2已經遍歷完了都能對應的上，返回true
        if (root2 == null) {
            return true;
        }
        //如果Tree2還沒有遍歷完，Tree1卻遍歷完了。返回false
        if (root1 == null) {
            return false;
        }
        //如果其中有一個點沒有對應上，返回false
        if (root1.val != root2.val) {
            return false;
        }
        //如果根節點對應的上，那麼就分別去子節點裡面匹配
        return doesTree1HaveTree2(root1.left, root2.left) && doesTree1HaveTree2(root1.right, root2.right);

    }

    public class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

        public TreeNode(int val) {
            this.val = val;

        }

    }
}