LeetCode Algorithm 572. 另一棵树的子树

​​572. 另一棵树的子树​​

Ideas

首先想到的就是递归判断两棵树的每一个节点是否相等，那么就需要将​​subRoot​​​跟​​root​​的每一个节点构成的子树判断是否相同。

递归判断相等的逻辑比较简单，首先当前两个节点值相等，然后递归判断左子树是否相等，再递归判断右子树是否相等。

递归的终止条件，就是两棵树同时到达了叶子结点，并且都没有孩子节点，此时说明这两颗子树相等。

再具体一点，我们首先要判断​​subRoot​​​是不是​​root​​的子树，有三种情况：

1. 1. 
      
2. ​​subRoot​​​和​​root​​两棵树相等；
3. ​​subRoot​​​是​​root​​的左子树；
4. ​​subRoot​​​是​​root​​的右子树。

然后我们要判断​​subRoot​​​和​​root​​是不是相等，需要满足三个条件：

1. 1. 
      
2. 当前两棵树的根节点相等；
3. ​​subRoot​​​的左子树和​​root​​的左子树相等；
4. ​​subRoot​​​的右子树和​​root​​的右子树相等；
   1. 
      

所以这里有两个递归的过程，我们需要再定义一个辅助函数。

Code

Python

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


class Solution:
    def isSubtree(self, root: TreeNode, subRoot: TreeNode) -> bool:
        if not root and not subRoot:
            return True
        if not root or not subRoot:
            return False
        return self.isSameTree(root, subRoot) or self.isSubtree(root.left, subRoot) or self.isSubtree(root.right, subRoot)

    def isSameTree(self, root: TreeNode, subRoot: TreeNode):
        if not root and not subRoot:
            return True
        if not root or not subRoot:
            return False
        return root.val == subRoot.val and self.isSameTree(root.left, subRoot.left) and self.isSameTree(root.right, subRoot.right)