【每日一題】二叉樹的層序遍歷2

題目:https://leetcode-cn.com/problems/binary-tree-level-order-traversal-ii/

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()) {
            int size = q.size();
            vector<int> cur_level;
            while (size--) {
                TreeNode* cur_node = q.front();  q.pop();
                cur_level.push_back(cur_node->val);
                if (cur_node->left) q.push(cur_node->left);
                if (cur_node->right) q.push(cur_node->right);
            }
            res.push_back(cur_level);
        }

        reverse(res.begin(), res.end());
        return res;
    }
};

EOF

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章