代码随想录算法训练营第十五天|226.翻转二叉树 101.对称二叉树 2-CSDN博客

本文链接：https://blog.csdn.net/minodeer/article/details/136973774

本文介绍了两个关于二叉树操作的问题：如何使用C++实现翻转二叉树（通过sweep方法），以及判断一个二叉树是否对称（通过递归函数compare）。

摘要由CSDN通过智能技术生成

226.翻转二叉树

class Solution {
public:
    void sweap(TreeNode* a, TreeNode* b) {
        TreeNode* cur = a;
        a = b;
        b = cur;
    }
    TreeNode* invertTree(TreeNode* root) {
        stack<TreeNode*> a;
        vector<int> ret;
        if (root != NULL)
            a.push(root);
        while (!a.empty()) {
            TreeNode* node = a.top();
            if (node != NULL) {
                a.pop();
                if (node->right)
                    a.push(node->right);
                if (node->left)
                    a.push(node->left);
                swap(node->left, node->right);
                a.push(node);
                a.push(NULL);
            } else {
                a.pop();
                TreeNode* k = a.top();
                a.pop();
                ret.push_back(k->val);
            }
        }
        return root;
    }
};

101.对称二叉树 2

class Solution {
public:
    bool compare(TreeNode* l, TreeNode* r) {
        if (l == NULL && r != NULL)
            return false;
        else if (l != NULL && r == NULL)
            return false;
        else if (l == NULL && r == NULL)
            return true;
        else if(l->val != r->val)
            return false;
        if (compare(l->left, r->right) && compare(l->right, r->left)) {
            return true;
        }
        return false;
    }
    bool isSymmetric(TreeNode* root) {
        if (root == NULL)
            return true;
        return compare(root->left, root->right);
    }
};