题干
572. 另一个树的子树(2020/5/7每日一题)
难度简单241收藏分享切换为英文关注反馈
给定两个非空二叉树 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。
思路
重点:对于树的操作,首先要考虑递归!
本题查找子树可以分解为两个子操作
- 判断右边树
t
是不是左边树s
的子树 - 判断右边树
t
是不是与左边树s
相等的树
若 s.val != t.val
则s
与t
一定不相同,继而判断t是否是s.left
或者s.right
的子树
若s.val == t.val
则s
与t
可能相同(只是可能),所以此时需要先判断s
是否与 t
相同,若相同,则返回True,若不相同,仍然需要判断t是否是s.left
或者s.right
的子树。
所以对于解决这道题,需要两个函数,
第一个是题目给定的判定子树的函数(对应操作1)
第二个是比较两棵树是否相同的函数(对应操作2)
- python代码
class Solution:
def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
if not t: return True
if not s: return False
if self.compare(s, t):
return True
return self.isSubtree(s.left, t) or self.isSubtree(s.right, t)
def compare(self, s, t):
if not s and not t: return True
if not (s and t): return False
if s.val != t.val:
return False
return self.compare(s.left, t.left) and self.compare(s.right, t.right)
当然,前面代码的逻辑只是为了方便阅读和理解,实际操作可以简化如下:
class Solution:
def isSubtree(self, s: TreeNode, t: TreeNode) -> bool:
if not t: return True
if not s: return False
return self.compare(s, t) or self.isSubtree(s.left, t) or self.isSubtree(s.right, t)
def compare(self, s, t):
if not s and not t: return True
if not (s and t): return False
return s.val == t.val and self.compare(s.left, t.left) and self.compare(s.right, t.right)