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] 范围内的整数。

解答

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public boolean flipEquiv(TreeNode root1, TreeNode root2) {
        return valuesAreEqual(root1, root2) && subtreesAreEqual(root1, root2);
    }

    // 已知node1与node2均不为空，且两者值相同时，判断两者是否翻转相同
    private boolean subtreesAreEqual(TreeNode node1, TreeNode node2) {
        if (node1 == null && node2 == null) return true;
        if (node1 == null && node2 != null || node1 != null && node2 == null) return false;
        if (valuesAreEqual(node1.left, node2.left) && valuesAreEqual(node1.right, node2.right))
            return subtreesAreEqual(node1.left, node2.left) && subtreesAreEqual(node1.right, node2.right);
        if (valuesAreEqual(node1.left, node2.right) && valuesAreEqual(node1.right, node2.left))
            return subtreesAreEqual(node1.left, node2.right) && subtreesAreEqual(node1.right, node2.left);
        return false;
    }

    // 判断两节点数值是否相同
    private boolean valuesAreEqual(TreeNode node1, TreeNode node2) {
        if (node1 == null && node2 == null) return true;
        if (node1 == null && node2 != null || node1 != null && node2 == null) return false;
        return node1.val == node2.val;
    }
}