572. 另一个树的子树
给定两个非空二叉树 s 和 t,检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。
示例 1:
给定的树 s:
3
/ \
4 5
/ \
1 2
给定的树 t:
4
/ \
1 2
返回 true,因为 t 与 s 的一个子树拥有相同的结构和节点值。
示例 2:
给定的树 s:
3
/ \
4 5
/ \
1 2
/
0
给定的树 t:
4
/ \
1 2
返回 false。
解题思路:和LeetCode-437. 路径总和 III思路相同。遇到树的问题尝试使用递归方法。isSubtreeHelper(self,s,t)方法是求解以s,t为根的两个二叉树是否相同。那么对原s二叉树所有节点使用这个方法,将所有结果取或运算即可得到答案。
Python3代码如下:
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def isSubtree(self, s, t):
"""
:type s: TreeNode
:type t: TreeNode
:rtype: bool
"""
if not s:
return False
return self.isSubtreeHelper(s,t) or self.isSubtree(s.left,t) or self.isSubtree(s.right,t)
def isSubtreeHelper(self,s,t):
if not s and not t:
return True
if not s or not t or s.val != t.val:
return False
return self.isSubtreeHelper(s.left,t.left) and self.isSubtreeHelper(s.right,t.right)