原地址
流程: 特例处理: 当根节点为空,则返回空列表 [] ; 初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ; BFS 循环: 当队列 queue 为空时跳出; 新建一个临时列表 tmp ,用于存储当前层打印结果; 当前层打印循环: 循环次数为当前层节点数(即队列 queue 长度); 出队: 队首元素出队,记为 node; 打印: 将 node.val 添加至 tmp 尾部; 添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ; 将当前层结果 tmp 添加入 res 。
?
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
if(root == nullptr) return vector<vector<int>>();
vector<vector<int>> vecRet;
queue<TreeNode *>que;
que.push(root);
while(!que.empty()) {
int size = que.size();
vector<int>vecTemp;
for(int i = 0; i < size; i++) {
//依次弹出并存值在temp数组中
auto node = que.front();
que.pop();
vecTemp.push_back(node->val);
if(node->left){
que.push(node->left);
}
if(node->right){
que.push(node->right);
}
}
vecRet.push_back(vecTemp);
}
return vecRet;
}
};
|