这道题一看会很头疼,可翻转的节点数很多,根本没法比,可能性也太多啦!
这再次体现了递归的强大,我们知道,二叉树问题除了广度遍历,树形dp,剩下的基本上都是递归解法。我们如何递归?对于两个根节点,如果皆为空那肯定比较成功。如果都不为空,如果其值不等,那一定比较失败,如果相等,我们应该比较其子节点,这是因为可能出现翻转情况,我们进行两次比较,一次翻转,一次不翻转,有一个成功即为成功。多余出来的情况会在第一次递归就结束返回false,也不会对性能有过多影响:
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
func flipEquiv(root1 *TreeNode, root2 *TreeNode) bool {
if root1 == nil && root2 == nil {
return true
}
if root1 == nil || root2 == nil {
return false
}
if root1.Val != root2.Val {
return false
}
compare1 := flipEquiv(root1.Left, root2.Left) && flipEquiv(root1.Right, root2.Right)
compare2 := flipEquiv(root1.Left, root2.Right) && flipEquiv(root1.Right, root2.Left)
return compare1 || compare2
}