【DFS】LeetCode 572. 另一棵树的子树

题目链接

572. 另一棵树的子树

思路

判断两个树是否相等的三个条件是与的关系，即：

1. 当前两个树的根节点值相等；
2. 并且，s 的左子树和 t 的左子树相等；
3. 并且，s 的右子树和 t 的右子树相等。

而判断 subRoot 是否为 root 的子树的三个条件是或的关系，即：

1. 当前两棵树相等；
2. 或者，t 是 s 的左子树；
3. 或者，t 是 s 的右子树。

代码

class Solution {
    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);
    }
}