【二叉树】翻转等价二叉树

原创

haocold 2022-10-24 19:20:42 博主文章分类：算法 ©著作权

文章标签 swift 算法二叉树子树子节点 文章分类 运维

©著作权归作者所有：来自51CTO博客作者haocold的原创作品，请联系作者获取转载授权，否则将追究法律责任

0x00 题目

我们可以为二叉树 T 定义一个翻转操作
选择任意节点，然后交换它的左子树和右子树
只要经过一定次数的翻转操作后
能使 X 等于 Y 我们就称二叉树 X 翻转 等价 于二叉树 Y

这些树由根节点 root1 和 root2 给出
如果两个二叉树是翻转 等价 的
则返回 true ，否则返回 false

0x01 思路

节点都为空时，相等 其中一个为空，或者值不相等时
则不相等
对于子节点，可以选择翻转
或者不翻转

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func flipEquiv(_ root1: TreeNode?, _ root2: TreeNode?) -> Bool {
    // 包含为 空 的情况
    if root1 === root2 { return true }
    
    // 只有一个为空，或者值不相等
    if root1 == nil || root2 == nil || root1?.val != root2?.val {
        return false
    }
    
    // 不翻转的情况
    let b1 = flipEquiv(root1?.left, root2?.left)
    let b2 = flipEquiv(root1?.right, root2?.right)
    
    // 翻转的情况
    let b3 = flipEquiv(root1?.left, root2?.right)
    let b4 = flipEquiv(root1?.right, root2?.left)
    
    return (b1 && b2) || (b3 && b4)
}