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

题目描述

思路

因为需要将每一层的数据对应放入一个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;
    }
};