572.另一棵树的子树

45 阅读1分钟

题目:
给你两棵二叉树 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)
}