leetcode 951. 翻转等价二叉树

思想

题目中说了，只要经过一定次数的翻转操作后，能使二叉树 X 等于二叉树 Y，我们就称二叉树 X 翻转等价于二叉树 Y，所以不一定在每一个root都要进行翻转。

上面的树的颜色分别代表递归思路（3）的①②③情况。

1. 标为黄色代表：$roo{t}_1$和$roo{t}_2$相比
2. 标为橘色代表以下两种情况都可以：
   1. $lef{t}_1$和$lef{t}_2$相比，$righ{t}_1$和$righ{t}_2$相比
   2. $lef{t}_1$和$righ{t}_2$相比，$righ{t}_1$和$lef{t}_2$相比

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool flipEquiv(TreeNode* root1, TreeNode* root2) {
        if(!root1 && !root2)
            return true;    // root1和root2都不存在，就说明当前节点相同
        bool flag = false;
        if(root1 && root2){
            if(root1->val == root2->val){   // root1和root2都存在，且值相同
                flag = true;    // 就说明当前节点是相同的
            }
            return flag && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right)) || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left));    
            // root1->val == root2->val代表当前节点相不相同 && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right))代表翻转后情况相不相同 
            // || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left))代表不翻转情况相不相同
        }
        return false;   // root1和root2其中有一个存在，另一个不存在就是false

    }
};

进一步观察我又进行了化简

class Solution {
public:
    bool flipEquiv(TreeNode* root1, TreeNode* root2) {
        if(!root1 && !root2)
            return true;    // root1和root2都不存在，就说明当前节点相同
        if(root1 && root2){
            return root1->val == root2->val && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right)) || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left));    
            // root1->val == root2->val代表当前节点相不相同 && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right))代表翻转后情况相不相同 
            // || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left))代表不翻转情况相不相同
        }
        return false;   // root1和root2其中有一个存在，另一个不存在就是false

    }
};