这是一道队列相关的例题
参考b站“懒猫老师”视频
伪代码
- 队列Q初始化
- 如果二叉树非空,将根指针入队
- 循环直到队列Q为空(循环的退出条件)
- q=队列Q的队头元素出队
- 访问节点q的数据域(这一步输出了)
- 若节点q存在左孩子,则将左孩子指针入队
- 若节点q存在右孩子,则将右孩子指针入队
这个方法每次弹出一个值 并压入它的所有子节点
完整代码参考
https://blog.csdn.net/ZM970307/article/details/79934150
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
typedef char TElemType;
typedef enum{ Link, Thread } PointerTag;
typedef struct BiTNode
{
TElemType data;
struct BiTNode* lChild;
struct BiTNode* rChild;
PointerTag lTag;
PointerTag rTag;
}BiTNode, *BiTree;
unsigned int TElemIndex = 0; //用于从数组创建树
/*从数组创建树*/
void createBiTreeByArray(BiTree *T, vector<TElemType> list)
{
if (TElemIndex + 1 <= list.size())
{
TElemType temp = list[TElemIndex];
if (temp == '#')
{
*T = NULL;
TElemIndex++; /*index必须使用全局变量++,
*而不能作为函数参数++(函数局部变量index在内层递归index++不会改变外层的index,退出内层递归时index会变回去)*/
}
else
{
*T = (BiTNode *)malloc(sizeof(BiTNode));
if (!(*T))
{
perror("fail: create-malloc");
return;
}
(*T)->data = temp;
(*T)->lChild = NULL;
(*T)->rChild = NULL;
(*T)->lTag = Link;
(*T)->rTag = Link;
TElemIndex++;
createBiTreeByArray(&(*T)->lChild, list);
createBiTreeByArray(&(*T)->rChild, list);
}
}
}
/*二叉树层序遍历*/
void levelTraverse(BiTree T)
{
queue<BiTree> q; //定义一个队列,数据类型是二叉树结点指针,不要仅是TElemType!不然无法遍历
if (T)
{
q.push(T);
}
while (!q.empty())
{
BiTree front = q.front(); //取队列最前面的结点
cout << front->data;
q.pop(); //删除最前面的结点
if (front->lTag == Link && front->lChild) //判断最前面的左结点是否为空,不是则放入队列
{
q.push(front->lChild);
}
if (front->rTag == Link && front->rChild) //判断最前面的右结点是否为空,不是则放入队列
{
q.push(front->rChild);
}
}
}
int main()
{
vector<TElemType> TElemList = { 'A', 'B', 'D', 'H', '#', 'K', '#', '#', '#', 'E', '#', '#', 'C', 'F', 'I', '#', '#', '#', 'G', '#', 'J', '#', '#' };
BiTree root = NULL;
createBiTreeByArray(&root, TElemList);
cout << "create ok." << endl;
cout << "层序遍历:";
levelTraverse(root);
cout << endl;
system("pause");
return 0;
}
leetcode官方解答改良了这里 每次弹出si个数 并压入它们的子节点 他的返回值是一个二维数组
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector <vector <int>> ret;
if (!root) {
return ret;
}
queue <TreeNode*> q;
q.push(root);
while (!q.empty()) {
int currentLevelSize = q.size();
ret.push_back(vector <int> ());
for (int i = 1; i <= currentLevelSize; ++i) {
auto node = q.front(); q.pop();
ret.back().push_back(node->val);
if (node->left) q.push(node->left);
if (node->right) q.push(node->right);
}
}
return ret;
}
};//这个的输出格式是二维数组 好像没办法用最基础的bfs?
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/binary-tree-level-order-traversal/solution/er-cha-shu-de-ceng-xu-bian-li-by-leetcode-solution/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
看到了一个写的更清晰的代码
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> res;
if(!root) return res;
queue<TreeNode*> qu;
qu.push(root);
while(!qu.empty())
{
vector<int> tmp;//这个里面放的是一层的节点
int len=qu.size();
for(int i=0;i<len;i++){
TreeNode* node=qu.front();
qu.pop();
tmp.push_back(node->val);
if(node->left) qu.push(node->left);
if(node->right) qu.push(node->right);
}
res.push_back(tmp);//把每一层的节点压进去
}
return res;
}
};