Leetcode 951.翻转等价二叉树（Flip Equivalent Binary Trees）

Leetcode 951.翻转等价二叉树

1 题目描述（Leetcode题目链接）

  我们可以为二叉树 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 的三个节点。

提示：

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

2 题解

  递归，具体见注释。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def flipEquiv(self, root1: TreeNode, root2: TreeNode) -> bool:
        if not root1 and not root2: # 如果两个root都为空则返回True
            return True
        if not root1 or not root2 or root1.val != root2.val: # 如果两个root只有一个为空，或者量值不相等则返回False
            return False
        a = self.flipEquiv(root1.left, root2.left) and self.flipEquiv(root1.right, root2.right) # 判断不交换左右孩子的子树是否相等
        b = self.flipEquiv(root1.left, root2.right) and self.flipEquiv(root1.right, root2.left) # 判断交换了左右孩子的子树是否相等
        return a or b