【Leetcode】102. Binary Tree Level Order Traversal

题目地址：

https://leetcode.com/problems/binary-tree-level-order-traversal/

树的广度优先遍历。

用一个队列即可。基本思想是，每当从队列中pop一个数出来的时候，就将它的子节点add进去，这样每一层遍历完后，队列里已经从左向右存储了当前这一层的所有子节点，于是就可以继续pop出来访问。需要注意的一个地方是，要提前存储队列的size，也就是当前这一层的节点数目，因为队列的size在pop的时候是动态变化的。代码如下：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left),
 * right(right) {}
 * };
 */
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.size()) {
      vector<int> v;
      for (int i = q.size(); i; i--) {
        auto t = q.front();
        q.pop();
        v.push_back(t->val);
        if (t->left) q.push(t->left);
        if (t->right) q.push(t->right);
      }
      res.push_back(v);
    }

    return res;
  }
};

时空复杂度$O(n)$。