从上到下按层打印二叉树(BFS详解(中等难度)C++)

tech2026-04-18  2

从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。

例如: 给定二叉树: [3,9,20,null,null,15,7],

3

/ 9 20 / 15 7 返回其层次遍历结果:

[ [3], [9,20], [15,7] ]

解: 解析都说用BFS遍历,先搞清楚啥是BFS遍历: BFS叫广度优先算法,全程是 breadth First Search,(也称宽度优先搜索)

这篇文章讲的相对详细一点:https://blog.csdn.net/g11d111/article/details/76169861

每次以扩散的方式向外访问顶点,和树的遍历一样需要使用一个队列,通过反复取出队首顶点,将该顶点可到达的未曾加入过队列的队列全部入队(这一步就是扩散),直到队列为空时遍历结束.

/** * 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搜索 //注意队列里面保存的是节点! queue<TreeNode*> q; //这个容器是用来返回数组用的 vector<vector<int>> res; //插入根节点,注意和set一样,都是push插入 q.push(root); while(!q.empty()) { //一定一定要在这里把q.size()赋给size int size=q.size(); //注意下面这个容器在每次大循环都会重新建立 vector<int> v; //如果这里使用q.size() //**那么每次循环q.size()都会改变的!!!** for(int i=0; i<size; i++) { //用一个临时节点保存队列中将要删除的节点 TreeNode* rt = q.front(); //把该节点弹出 q.pop(); //要是空节点,就不用下面的插入值和左右分支了 //直接开始下一次循环,!rt 和rt==NULL没区别 if(!rt) continue; //把弹出节点的值保存到v容器中 v.push_back(rt->val); //因为队列先进先出,所以先把左边插入 if(rt->left) q.push(rt->left); if(rt->right) q.push(rt->right); } //每次插入一个容器 if(v.size()!=NULL) res.push_back(v); } return res; } };
最新回复(0)