leetcode刷题(剑指offer) 572.另一棵树的子树

572.另一棵树的子树

给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在，返回 true ；否则，返回 false 。

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

示例 1：

输入：root = [3,4,5,1,2], subRoot = [4,1,2]
输出：true

示例 2：

输入：root = [3,4,5,1,2,null,null,null,null,0], subRoot = [4,1,2]
输出：false

提示：

• root 树上的节点数量范围是 [1, 2000]
• subRoot 树上的节点数量范围是 [1, 1000]
• -104 <= root.val <= 104
• -104 <= subRoot.val <= 104

题解：

可以先将题目进行简化，变成如何判断两棵树相等。从根节点开始，如果根节点相等，那就递归的比较左子树和右子树。

代码实现如下：

private boolean isSametree(TreeNode root, TreeNode subRoot) {
    if (root == null && subRoot == null) {
        return true;
    }
    if (root == null || subRoot == null) {
        return false;
    }
    return root.val == subRoot.val && isSametree(root.left, subRoot.left) && isSametree(root.right, subRoot.right);
}

在实现了isSametree的基础上，判断一棵树是否为另一颗树的一部分，需要将root和subRoot先放入isSametree中进行比较，然后再将root的左子树和右子树分别递归调用isSubtree即可。

代码实现如下：

public boolean isSubtree(TreeNode root, TreeNode subRoot) {
    if (root == null && subRoot == null) {
        return true;
    }
    if (root == null || subRoot == null) {
        return false;
    }
    return isSametree(root, subRoot) || isSubtree(root.left, subRoot) || isSubtree(root.right, subRoot);
}

private boolean isSametree(TreeNode root, TreeNode subRoot) {
    if (root == null && subRoot == null) {
        return true;
    }
    if (root == null || subRoot == null) {
        return false;
    }
    return root.val == subRoot.val && isSametree(root.left, subRoot.left) && isSametree(root.right, subRoot.right);
}