【每日一题】LeetCode. 102.二叉树的层次遍历

每日一题，防止痴呆 = =

一、题目大意

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal

二、题目思路以及AC代码

这题就很常规的题，LeetCode居然分类为中等，确实这个分类有点迷，反正就当复习一下以前的知识吧。

思路

要按层进行遍历，并且存储下来，只需要在BFS的时候，每次先计算一下该层有多少个元素，然后只遍历这些元素，并且把新得到的下层的元素入队即可。

下面给出AC代码：

/**
 * 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>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if (!root) return res;

        queue<TreeNode*> q;
        q.push(root);

        while (!q.empty()) {
            int len = q.size();

            res.push_back(vector<int> ());
            for (int i=0;i<len;i++) {
                TreeNode* cur = q.front();  q.pop();

                res.back().push_back(cur->val);

                if (cur->left) q.push(cur->left);
                if (cur->right) q.push(cur->right);
            }
        }

        return res;
    }
};

如果有问题，欢迎大家指正！！！