从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
题源链接:https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof
解题思路:
- 按层打印: 题目要求的二叉树的 从上至下 打印(即按层打印),又称为二叉树的 广度优先搜索(BFS)。BFS 通常借助 队列 的先入先出特性来实现。
- 每层打印到一行: 将本层全部节点打印到一行,并将下一层全部节点加入队列,以此类推,即可分为多行打印。
算法流程:- 特例处理: 当根节点为空,则返回空列表 [] ;
- 初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
- BFS 循环: 当队列 queue 为空时跳出;
- 新建一个临时列表 tmp ,用于存储当前层打印结果;
- 当前层打印循环: 循环次数为当前层节点数(即队列 queue 长度);
- 出队: 队首元素出队,记为 node;
- 打印: 将 node.val 添加至 tmp 尾部;
- 添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
- 将当前层结果 tmp 添加入 res 。
- 返回值: 返回打印结果列表 res 即可。
流程图连接
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
/* BFS,广度优先搜索算法进行tree遍历(借助queue)*/
// 声明二维数组存储每一层节点的值
vector<vector<int>> result;
// 声明一个队列保存tree每个节点(先进先出)
queue<TreeNode*> nodes;
// 将根节点最先入列
if (root != NULL){
nodeQueue.push(root);
}
// 判断队列里面是否有元素
while(nodes.empty() == false){
// 声明临时数组,保存每一层每个节点的值
vector<int> temp_result;
int layerNodeNums = nodes.size();
for (int i=0; i<layerNodeNums; i++){
// 获取队首元素
TreeNode *currentNode = nodes.front();
// 删除队首元素, 用完就出列
nodes.pop();
temp_result.push_back(currentNode->val);
// 将此节点的左右节点入列
if (currentNode->left){
nodes.push(currentNode->left);
}
if(currentNode->right){
nodes.push(currentNode->right);
}
}
// 将每一层的元素列表保存到result中
result.push_back(temp_result);
// 清空临时vector,并回收内存
vector<int>().swap(temp_result);
}
return result;
}
};