力扣-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;}
};

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