题解 | #求二叉树的层序遍历#
求二叉树的层序遍历
https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
class Solution {
public:
/**
*
* @param root TreeNode类
* @return int整型vector<vector<>>
*/
vector<vector<int> > levelOrder(TreeNode* root) {
vector<vector<int>> result; // 创建结果数组
queue<TreeNode*> que; // 创建单向队列,存放每层的结点
if(root) que.push(root); // 如果根节点不为空,将根节点放入队列中
while(!que.empty()){ // 当队列不为空时,进入循环
vector<int> temp_vec; // 创建临时数组,保存每层结点值
int deq_size = que.size(); // 保存每层结点数,此为每层遍历次数
for(int i = 0; i < deq_size; ++i){ // 进入循环
TreeNode* cur = que.front(); // 临时结点等于队头结点
temp_vec.push_back(cur->val); // 将队头结点值放入临时数组中
que.pop(); // 删除队头结点
if(cur->left) que.push(cur->left); // 当左孩子不为空,将左孩子放入队列中
if(cur->right) que.push(cur->right); // 当右孩子不为空,将右孩子放入队列中
}
result.push_back(temp_vec); // 将该层结点值放入结果数组中
}
return result;
}
};
腾讯云智研发成长空间 229人发布
