leetcode_951 翻转等价二叉树

要求

我们可以为二叉树 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] 范围内的整数。

核心代码

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
        
class Solution:
    def flipEquiv(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> bool:
        return self.check(root1,root2)
    
    def check(self,root1,root2):
        if not root1:
            if not root2:
                return True
            return False
        elif not root2:
            return False
        
        if root1.val != root2.val:
            return False
        
        if self.check(root1.left,root2.left) and self.check(root1.right,root2.right):
            return True
        
        if self.check(root1.left,root2.right) and self.check(root1.right,root2.left):
            return True
        
        return False

解题思路：题目中的比较关键的点：选择任意节点，然后交换它的左子树和右子树,也就是说我们的在某些节点上可以翻转也可以不动，所以我们递归进行检查是不是等价的时候，我们既要检查同左子树，又要检查一左一右的情况。