劍指offer-17:樹的子結構

題目描述

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

解題思路

• 從根節點開始，判斷值是否相同
• 如果不同則分別用A樹的左右孩子節點與B樹的根節點比較
• 沒有則直接返回
• 如果找到值相同的，則呼叫輔助方法遞迴判斷A樹的左右孩子和B樹的左右孩子是否相等
• 直到B樹為空，則返回true，否則回溯，繼續尋找A樹的與B跟節點值相同的孩子
• 直到A樹為空，則返回false
• 注意邊界判斷。空樹不是任意一個樹的子結構

程式碼實現

/**
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 res = false;
        if (root1 == null || root2 == null) {
            return false;
        }
        if (root1.val == root2.val) {
            res = HasSubTreeHelp(root1, root2);
        }
        if (!res) {
            res = HasSubTreeHelp(root1.left, root2);
        }
        if (!res) {
            res = HasSubTreeHelp(root1.right, root2);
        }
        return res;
    }
    
    public boolean HasSubTreeHelp (TreeNode root1,TreeNode root2) {
        if (root2 == null) return true;
        if (root1 == null) return false;
        if (root1.val == root2.val){
            return HasSubTreeHelp(root1.left, root2.left) 
            	&& HasSubTreeHelp(root1.right, root2.right);
        }
        return false;
    }
}