上一篇文章主要讲了二叉树的前中后序遍历算法的递归非递归实现,本文介绍二叉树的另一种遍历方式——层序遍历。
层序遍历就是从所在二叉树的根节点出发,首先访问第一层的树根节点,然后从左到右访问第二层上的节点,接着是第三层的节点,以此类推,自上而下,自左至右逐层访问树的结点的过程。
具体使用一个先进先出的队列来实现,基本过程为:
- 根节点入队,然后:
- 从队列中取出一个节点
- 访问该节点
- 若该节点所指节点的左、右子树非空,则将其左右子树顺序入队
首先实现一个队列,采用链式存储(队列的链式存储结构,其实就是线性表的单链表,只不过它只是尾进头出而已),相较于顺序存储更灵活。
//树节点定义
#include <iostream>
#include <memory>
#include <vector>
typedef int ElementType;
struct TreeNode {
TreeNode() :lchild(nullptr), rchild(nullptr) {
};
TreeNode(ElementType d, std::shared_ptr<TreeNode> lhs, std::shared_ptr<TreeNode> rhs) :
data(d), lchild(lhs), rchild(rhs) {
};
ElementType data;
std::shared_ptr<TreeNode> lchild;
std::shared_ptr<TreeNode> rchild;
};
using BinTree = std::shared_ptr<TreeNode>;
//队列类
class MyQueue {
//队列存储元素
struct QNode {
BinTree data;
std::shared_ptr<QNode> next;
QNode() :next(nullptr) {
};
QNode(BinTree d, std::shared_ptr<QNode> p) :
data(d), next(p) {
};
};
using PQNode = std::shared_ptr<QNode>;
public:
MyQueue(); //构造函数
void addQ(BinTree bt); //入队
PQNode deleteQ(); //出队
bool empty(); //判断队列空
//bool full(); //判断队列满
private:
PQNode Front, Rear; //队列的头尾指针
//int MaxSize; //设置队列最大容量,毕竟内存空间是有限的,配合full()函数,在这暂不考虑内存空间耗尽情况
};
MyQueue::MyQueue() :Front(new QNode), Rear(Front) {
}; //构造函数
void MyQueue::addQ(BinTree bt) //入队
{
//尾部入队
PQNode newNode = std::make_shared<QNode>();
newNode->data = bt;
newNode->next = nullptr;
Rear-</