Note:
开个指针记录每一层的最后一个节点的位置,当队头指针访问到这一层最后一个节点的位置以后就更新一下这个指针,让他继续指向下一层最后一个节点的位置。
因为当队头遍历到这一层最后一个节点的位置以后,下一层的也就Push完了
过程中就是用vector来模拟了一下队列,方便使用下标去定每一层最后一个节点的指针
代码如下:
/**
* 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) {
if(!root) return {};
vector<vector<int>> ans;
vector<int> path;
vector<TreeNode*> q;
q.push_back(root);
int f = -1, t = 0, last = 0;
while(f < t){
auto node = q[++ f];
path.push_back(node -> val);
if(node -> left) q.push_back(node -> left), t ++;
if(node -> right) q.push_back(node -> right), t ++;
if(last == f){
ans.push_back(path);
path.clear();
last = t;
}
}
return ans;
}
};