leetcode - 4 翻转等价二叉树

翻转等价二叉树

我们可以为二叉树 T 定义一个翻转操作，如下所示：选择任意节点，然后交换它的左子树和右子树。

只要经过一定次数的翻转操作后，能使 X 等于 Y，我们就称二叉树 X 翻转等价于二叉树 Y。

编写一个判断两个二叉树是否是翻转等价的函数。这些树由根节点 root1 和 root2 给出。

示例：

输入：root1 = [1,2,3,4,5,6,null,null,null,7,8], root2 = [1,3,2,null,6,4,5,null,null,null,null,8,7]
输出：true
解释：我们翻转值为 1，3 以及 5 的三个节点。
Flipped Trees Diagram

提示：

每棵树最多有 100 个节点。
每棵树中的每个值都是唯一的、在 [0, 99] 范围内的整数。

因为是限时做的，所以比较匆忙，做的很乱

我这个题目做的比较潦草，中心思想就是确认 左对右 还是 左对左 之后正常 dfs 即可。

但是！这里我写的这么复杂主要是因为要判断左右孩子对齐问题

那么如果我们换一种想法呢？比如不管左右都做，然后判断哪一个对，这样在代码上会简介很多

考场上限时的时候写的：

def flipEquiv(self, root1: TreeNode, root2: TreeNode) -> bool:
    def dfs(root1,root2):
        if not root1 and root2:
            return False
        elif not root2 and root1:
            return False
        elif not root2 and not root1:
            return True
        flag = True
        leftval1,leftval2,rightval1,rightval2 = 0,0,0,0
        if root1.left:
            leftval1 = root1.left.val
        if root2.left:
            leftval2 = root2.left.val
        if root2.right:
            rightval2 = root2.right.val
        if root1.right:
            rightval1 = root1.right.val
        if (leftval1 == leftval2) and (rightval1 == rightval2):
            flag = dfs(root1.left,root2.left) and dfs(root1.right,root2.right)
        elif (leftval1 == rightval2 and rightval1 == leftval2):
            flag = dfs(root1.left,root2.right) and dfs(root1.right,root2.left)
        else:
            return False
        return flag
    return dfs(root1,root2)

时间复杂度为 O(min(N1,N2)), 空间复杂度为 O(min(H1​,H2​))

更好的修改方案：

def flipEquiv(self, root1, root2):
    def dfs(root1,root2)
        if root1 is root2:
            return True
        if not root1 or not root2 or root1.val != root2.val:
            return False

        return (dfs(root1.left, root2.left) and
                dfs(root1.right, root2.right) or
                dfs(root1.left, root2.right) and
                dfs(root1.right, root2.left))
    return dfs(root1,root2)

时间复杂度为 O(min(N1,N2)), 空间复杂度为 O(min(H1​,H2​))