力扣-102.二叉树的层序遍历

题目描述

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

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode *root) {
        vector<vector<int>> res;
        if (!root)
            return res;
        vector<int> level; //收集一层的
        queue<TreeNode *> q;
        q.push(root);
        while (!q.empty()) {
            int size = q.size();
            for (int i = 0; i < size; ++i) {
                TreeNode *tmp = q.front();
                q.pop();
                level.push_back(tmp->val);
                if (tmp->left != NULL)
                    q.push(tmp->left);
                if (tmp->right != NULL)
                    q.push(tmp->right);
            }
            res.push_back(level);
            level.clear();
        }
        return res;
    }
};

小结：通过这道题复习了一下层序遍历的思想：用一个队列先放入根结点，每次出队时左右孩子分别进队，最后输出顺序就是层序遍历。难点是如何判断一层的结束，这里使用的方法是记录队列的大小，每遍历一层更新一次。