题目:
解答:
上一题采用的深搜+记录深度的方法完成的,这题采用广搜的方法完成。
采用一个队列,遍历时将每一层的节点加入到队列中,如何知道每一层多少个节点呢。
这里采用一个变量size,代表每一层的节点个数。
遍历的时候,遍历size个节点,将节点加入到结果res中,同时,将子节点再加入到queue中,遍历到最后,上一层的节点已经全部删除掉,queue中存的是下一层的节点。
所以在每次遍历的开始,queue中存的是全部当前节点,遍历过程中不断加入下一节点,遍历size个节点之后,本次遍历结束, queue中存放的是下一层的所有节点,size值为下一层所有节点的个数。
开始的时候,将root加入到queque中,size = 1,代表第一层有一个节点。
如果queque为空,那么遍历结束。
最后,翻转vector获得最终结果, 采用函数reverse(res.begin(), res.end());
代码:
class Solution {
public:
vector<vector<int> > levelOrderBottom(TreeNode *root) {
vector<vector<int> > res;
if (root == NULL)
return res;
queue<TreeNode *> que;
que.push(root);
vector<int> row;
while (!que.empty())
{
row.clear();
int size = que.size();
while (size)
{
TreeNode *node = que.front();
que.pop();
row.push_back(node->val);
if (node->left) que.push(node->left);
if (node->right) que.push(node->right);
--size;
}
res.push_back(row);
}
reverse(res.begin(), res.end());
return res;
}
};