C++详解Leetcode:102. Binary Tree Level Order Traversal

原题

思想

代码和详解

struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
};

class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
//分配一个二维向量
vector<vector<int>> result;
//用队列来存储二叉树结点值
queue<TreeNode *> q;
//如果根节点为空，则返回
if (!root)
{
return result;
}
//将根节点先放入队列中
q.push(root);
//再将NULL放入队列中，NULL是为了标记那一层的作用
q.push(NULL);

//cur是为了存储每一层的节点值
vector<int> cur;
//循环进行二叉树层次遍历，直到队列为空
while (!q.empty())
{
//去除队列的第一个值
TreeNode* t = q.front();
//出队列
q.pop();

//如果t==NULL则证明当前层的遍历已经结束，需要将这一层存在cur中的值放到result中
if (t == NULL)
{
result.push_back(cur);
//将cur大小重新置为0
cur.resize(0);
//如果队列不为空，证明还没有遍历结束，继续讲NULL入队列作为标记
if (q.size() > 0)
{
q.push(NULL);
}
}
else
{
//将当前的节点值存到cur中
cur.push_back(t->val);
//如果左孩子不为空，则入队列
if (t->left)
{
q.push(t->left);
}
//如果右孩子不为空，则入队列
if (t->right)
{
q.push(t->right);
}
}
}
//返回结果
return result;
}
};