102. 二叉树的层序遍历

问题描述

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

分析

对于nullptr：

• 先判不空再入队
• 入队列后在for中判空，为空则continue

第一种更好，因为如果为空，即使continue也会影响全局，比如该题中res.push_back(layer_res);当某层结点为空，则layer_res为空，res会多了一个nullptr。当然可以处理，不过显然第一种更简洁。

bfs解法

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int> > res;
        queue<TreeNode*> q;
        if (root != nullptr) {
            q.push(root);
        }
        while(!q.empty()) {
            int layer_size = q.size();
            vector<int> layer_res;
            for (int i = 0; i < layer_size; i++) {
                TreeNode* t = q.front();
                q.pop();
                layer_res.push_back(t->val);
                if (t->left != nullptr) {
                    q.push(t->left);
                }
                if (t->right != nullptr) {
                    q.push(t->right);
                }
            }
            res.push_back(layer_res);
        }
        return res;
    }
};