题目:
给你两棵二叉树 root 和 subRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false 。
二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。
算法:
func isSubtree(root *TreeNode, subRoot *TreeNode) bool {
var dfs func(node *TreeNode) bool
dfs = func(node *TreeNode) bool {
if node == nil {
return false
}
// 如果node.Val == subRoot.Val不能直接return equal(node, subRoot),因为可能在左右子树才存在相等的子树
return equal(node, subRoot) || dfs(node.Left) || dfs(node.Right)
}
return dfs(root)
}
func equal(a, b *TreeNode) bool {
if a == nil && b == nil {
return true
}
if a == nil || b == nil {
return false
}
if a.Val != b.Val {
return false
}
return equal(a.Left, b.Left) && equal(a.Right, b.Right)
}