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

提示：

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

 

一开始代码是这样的，判断左右子节点是否相等，不相等就进行swap，需要注意的时空节点的判断。因为要调用swap，所以引入了对空节点的判断。另外swap也是明显不合理，怎么能改变tree的架构。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool flipEquiv(TreeNode* root1, TreeNode* root2) {
        bool ans = true;
        DFS(root1,root2,ans);
        return ans;
    }
    void DFS(TreeNode* root1, TreeNode* root2,bool &ans){
        if((root1&&root2)&&(root1->val!=root2->val)){
            ans = false;
            return;
        }
        if(root1==NULL&&root2==NULL) return;
        if(root1==NULL ^root2==NULL) {
            ans = false;
            return;
        }
        bool left_eq_right = (root1->left == NULL &&root2->right == NULL) ||
        ((root1->left != NULL &&root2->right != NULL)&&(root1->left->val==root2->right->val));
        bool right_eq_left = (root2->left == NULL &&root1->right == NULL) ||
        ((root2->left != NULL &&root1->right != NULL)&&(root2->left->val==root1->right->val));

        bool left_eq_left  = (root1->left == NULL && root2->left == NULL) ||
        ((root1->left != NULL &&root2->left != NULL)&&(root1->left->val==root2->left->val));
        bool right_eq_right = (root2->right == NULL &&root1->right == NULL) ||
        ((root1->right != NULL &&root2->right!= NULL)&&(root1->right->val==root2->right->val));

        if(left_eq_right&&right_eq_left){
            swap(root2->left,root2->right);
        }else if(!(left_eq_left&&right_eq_right)){
            ans = false;
            return;
        }
        if(root1->left) DFS(root1->left,root2->left,ans);
        if(root1->right) DFS(root1->right,root2->right,ans);
        
    }
};

参考其他人的代码之后的代码，

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool flipEquiv(TreeNode* root1, TreeNode* root2) {
        if((root1==NULL)||(root2==NULL)) return root1==root2;

        if(root1->val!=root2->val) return false;
        
        return (flipEquiv(root1->left,root2->left)&&flipEquiv(root1->right,root2->right))||
                (flipEquiv(root1->left,root2->right)&&flipEquiv(root1->right,root2->left));
    
    }

};