题目
给定两个非空二叉树 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。
思路
思路1:本题可以通过将树的空节点分类补全添加到原树上,如图(5没补)
代码如下:
# Definition for a binary tree node.
# class TreeNode:
# 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
"""
def convert(p):
return "^" + str(p.val) + "#" + convert(p.left) + convert(p.right) if p else "$"
return convert(t) in convert(s)
运行结果:
input:
[3,4,5,1,2]
[4,1,2]
树形表示:
3
/ \
4 5
/ \
1 2
4
/ \
1 2
打印的t:
^4#^1#$$^2#$$
打印的s:
^3#^4#^1#$$^2#$$^5#$$
思路2:暴力方法,遍历一遍大树s,将每个节点和t的根节点作比较,相等的话再遍历t去一个个比较。代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def isSubtree(self, a, b):
if not b:
return True
def checkTree(root1, root2):
if not root1 and not root2:
return True
elif root1 and not root2 or root2 and not root1:
return False
if root1.val != root2.val:
return False
return checkTree(root1.left, root2.left) and checkTree(root1.right, root2.right)
def dfs(s, t):
if not s:
return False
if s.val == t.val and checkTree(s, t):
return True
return dfs(s.left, t) or dfs(s.right, t)
return dfs(a, b)