LeetCode 101 对称二叉树

最新推荐文章于 2024-06-22 22:59:06 发布

午夜&星空

最新推荐文章于 2024-06-22 22:59:06 发布

阅读量98

点赞数

文章标签： leetcode 算法职场和发展 c++

本文链接：https://blog.csdn.net/SYX6663/article/details/129011300

版权

题目内容：给你一个二叉树的根节点 root ，检查它是否轴对称。

思路1:首先想到了中序遍历输出数组，然后根据数组是否对称判断树是否对称。


class Solution {
public:
    void inOrder(TreeNode* root,vector<int>& a){       
        if(root==nullptr){
            a.push_back(1001);
            return;
        }
                       
        if(root->left!=nullptr||root->right!=nullptr)inOrder(root->left,a);
        a.push_back(root->val);
        if(root->left!=nullptr||root->right!=nullptr)inOrder(root->right,a);                      
    }
    
    bool isSymmetric(TreeNode* root) {
        vector<int> a;
        inOrder(root,a);               
        int mid=a.size()/2;
        for(int i=0;i<a.size();i++){
            cout<<a[i];
        }
        if(mid==1&&a[0]!=a[2])return false;
        for(int i=1;i<=mid;i++){
            if(a[mid+i]!=a[mid-i])return false;
            if(a[mid+i]==1001&&a[mid-i]!=1001)return false;
            if(a[mid+i]!=1001&&a[mid-i]==1001)return false;
        }
        return true;
    }
};

对中序遍历做了一些修改，只输出存在的节点以及兄弟节点非空的空节点。由于val值小于1000，所以在得到的数组中用1001表示空节点。

但是由该方法遍历得到的数组对称并不能完全保证树的对称。如[5,4,1,null,1,null,4,2,null,2,null]。

于是放弃这种方法，参考题解学习使用递归法。

思路2：如何判断左右子树对称呢？

需要确保：1.左右节点数值相等或都为空节点。2.左节点的右子树与右节点的左子树对称。3.左节点的左子树与右节点的右子树对称。

可以发现，判断左右子树对称需要判断他们的子树之间是否对称。

class Solution {
public:
    bool symmetric(TreeNode* rootA,TreeNode* rootB){       
        if(rootA==nullptr&&rootB!=nullptr) return false;
        if(rootA!=nullptr&&rootB==nullptr) return false;
        if(rootA==nullptr&&rootB==nullptr) return true;       
        return symmetric(rootA->left,rootB->right)&&symmetric(rootB->left,rootA->right)&&rootA->val==rootB->val;
    }  //判断两颗树是否对称
    bool isSymmetric(TreeNode* root){
        if(root==nullptr){ 
            return true;
        }
        return symmetric(root->left,root->right);
    }
};