二叉树的层次遍历 II[LeetCode-107]

最新推荐文章于 2025-08-23 16:01:06 发布

转载最新推荐文章于 2025-08-23 16:01:06 发布 · 64 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://juejin.im/post/5c8fac85e51d457d006e8419

文章标签：

#数据结构与算法

博客围绕数据结构与算法展开，针对题目，提出将每一层数据对应放入vector的思路。遍历某层元素时，把对应子节点放入容器，利用queue先进先出特性存放节点。输出时，顶层元素放最后，采用vector的insert方法而非push_back。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

思路

因为需要将每一层的数据对应放入一个vector中，因此在遍历某一层元素时，同时将对应该层元素的所有子节点放入容器中，
又因为每一层的元素是从左到右依次放入vector，直接使用queue来存放各个节点即可，先进先出刚好满足这个特性。
因为输出的时候，顶层的元素放最后，因此不要用vector的push_back，最好用vector的insert(vector.begin(), element)

实现如下

/**
 * 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:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>> res;
        if (!root) {
            return res;
        }
        
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()) {
            vector<int> tmp;
            int size_level = q.size(); // 这里记录本层的元素数，下面for循环不会遍历下一层元素
            for (int i = 0; i < size_level; i++) {
                TreeNode* node = q.front();
                tmp.push_back(node->val);

                if (node->left) q.push(node->left); // 顺便把子节点也放入队列，等下一个循环再继续处理
                if (node->right) q.push(node->right);
                q.pop();
            }
            res.insert(res.begin(), tmp); // 这样便可实现下层的节点可以在输出的时候放在最前面
        }
        return res;
    }
};
复制代码