二叉树的遍历方法除了有前序遍历、中序遍历和后序遍历之外,还有第四种方式层序遍历。
1 层序遍历简介
层序遍历的含义是二叉树按照从根节点到叶子节点的层次关系,一层一层横向遍历各个节点。如图1所示的二叉树,按照层序遍历方法输出的结果为“1->2->3->4->5->6”。
图1 二叉树
所以,广度优先搜索一般使用层序遍历。
2 层序遍历的流程
层序遍历的流程如图2所示。
图2 层序遍历的流程
相关链接1 队列,queue,是C++ STL中定义的简单数据结构,它以先入先出有(FIFO)的方式插入数据,队列中的数据以连续的方式存储,只能访问队列中的第一个和最后一个元素。只能在队列的末尾添加新元素,只能从队列的头部移除元素。
根据图1所示的二叉树和图2所示的流程图,可得到图3所示的输出与队列的对应关系。其中,节点1首先入队列,此时队列中的数据为节点1;之后节点1出队列并输出;接下来将节点1的左子树节点2和右子树节点3入队列;然后节点2出队列并输出;之后节点2的左子树节点4和右子树节点5入队列,此时队列中的数据为节点3、节点4和节点5;接下来节点3出队列并输出;之后将节点3的右子树节点6入队列,此时队列中的数据为节点4、节点5和节点6;最后将点4、节点5和节点6依次出队列并输出。
图3 输出与队列的对应关系
3 代码实现
通过层序遍历的代码如下所示。
void levelOrderTraversal(TreeNode* root)
{
queue<TreeNode*> my_queue;
my_queue.push(root);
while (!my_queue.empty())
{
TreeNode* node = my_queue.front();
my_queue.pop();
cout << node->data<<endl;
if (node->leftChild != nullptr)
{
my_queue.push(node->leftChild);
}
if (node->rightChild != nullptr)
{
my_queue.push(node->rightChild);
}
}
}
其中,my_queue是C++中的queue类型的对象,即队列。通过my_queue调用front()函数,获取队列头部的数据,调用pop()函数令头部数据出队列。