算法训练Day13 | 二叉树层序遍历、226.翻转二叉树、101. 对称二叉树

最新推荐文章于 2024-06-21 17:14:10 发布

WIIIter

最新推荐文章于 2024-06-21 17:14:10 发布

阅读量250

点赞数 6

分类专栏： LeetCode刷题文章标签：算法

本文链接：https://blog.csdn.net/WIIIter/article/details/136923297

版权

LeetCode刷题专栏收录该内容

49 篇文章 0 订阅

订阅专栏

层序遍历

层序遍历就是从左到右一层一层的去遍历二叉树，这也是一种常用的二叉树的存储结构。
这个可以借用一个队列来实现层序遍历，队列先进先出，符合一层一层遍历的逻辑。简单来说，首先将根节点放入队列，此时队列里存储的就是第一层的节点，之后把队列里的节点取出的同时将该节点的左右子树按顺序放入队列中，这样就可以实现逐层遍历节点。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            // 这里一定要使用固定大小size，不要使用que.size()，因为que.size是不断变化的
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        return result;
    }
};

226.翻转二叉树

题目链接：226.翻转二叉树 -力扣（LeetCode）

思路：很直觉的想法，遍历每一个节点，将每个节点的左右子树翻转，这里我们用递归实现

class Solution {
public:
    void inver(TreeNode* node){
        if(!node) return;
        TreeNode* left = node->left;
        node->left = node->right;
        node->right = left;
        inver(node->left);
        inver(node->right);
    }
    TreeNode* invertTree(TreeNode* root) {
        inver(root);
        return root;

    }
};

参考文章：代码随想录-226.翻转二叉树

101. 对称二叉树

题目链接：101. 对称二叉树-力扣（LeetCode）

思路：对于二叉树是否对称，要比较的是根节点的左子树与右子树是不是相互翻转的，我们这里通过“后序”遍历去判断两颗子树是否是对称的

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

    }
    bool isSymmetric(TreeNode* root) {
        if(!root) return true;
        return Symm(root->left, root->right);
    }
};