剑指 Offer 26. 树的子结构

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构，即A中有出现和B相同的结构和节点值。

例如: 给定的树A:

给定的树 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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

递归搜索。B为空匹配成功，A为空匹配失败。若B不为根，当且节点又匹配失败，直接返回false。若B与当前节点匹配成功，那么A的左右孩子与B的左右孩子一定匹配成功，或者B与当前节点的左右子树匹配成功。

class Solution {

    TreeNode root = null;

    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if (B == null) return false;
        this.root = B;
        return dfs(A, B);
    }

    boolean dfs(TreeNode A, TreeNode B) {
        if (B == null) return true;
        if (A == null) return false;
        // 若B不为根，则当前必须匹配成功
        if (A.val != B.val && B != root) return false;
        // 要么B与当前节点匹配成功，要么与当前节点的左右子树匹配成功
        return A.val == B.val && dfs(A.left, B.left) && dfs(A.right, B.right) || dfs(A.left, B) || dfs(A.right, B);

    }
}

复杂度

时间复杂度：A的深度乘以B的深度。
空间复杂度：树高。