算法 | 《劍指offer》面試題26. 樹的子結構

輸入兩棵二叉樹A和B，判斷B是不是A的子結構。(約定空樹不是任意一個樹的子結構)

B是A的子結構， 即 A中有出現和B相同的結構和節點值。

例如:
給定的樹 A:
​ 3
​ /
4 5
/
1 2
給定的樹 B：
4
/
1
返回 true，因爲 B 與 A 的一個子樹擁有相同的結構和節點值。

示例 1：

輸入：A = [1,2,3], B = [3,1]
輸出：false

示例 2：

輸入：A = [3,4,5,1,2], B = [4,1]
輸出：true

限制：

0 <= 節點個數 <= 10000

題解：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(A == null || B == null)  return false;
        return dfs(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }
    private boolean dfs(TreeNode A, TreeNode B) {
        if(A == null)  return false;
        if(B == null)  return true;
        return A.val == B.val && dfs(A.left, B.left) && dfs(A.right, B.right);
    }
}