跳转至

LeetCode: 951. 翻转等价二叉树

1、题目描述

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

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

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

示例:

输入: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
解释:We flipped at nodes with values 1, 3, and 5.

Flipped Trees Diagram

提示:

  • 每棵树最多有 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:
            return True
        if (not root1 and root2) or (root1 and not root2) or root1.val != root2.val:
            return False

        return ((self.flipEquiv(root1.left, root2.right) and
                 self.flipEquiv(root1.right, root2.left)) or
                (self.flipEquiv(root1.left, root2.left) and
                 self.flipEquiv(root1.right, root2.right)))