在《C++中二叉树的递归非遍历方法3-3》中提到,二叉树的层序遍历的输出顺序是从根节点开始,一层一层横向遍历各个节点。如图1所示的二叉树,层序遍历的输出的输出顺序为“1->2->3->4->5->6”。
图1 二叉树结构
1 递归实现层序遍历的流程
递归实现层序遍历的流程如图2所示。
图2 递归实现层序遍历的流程
在图2所示的流程中,除了判断当前节点是否为空外,还需要判断当前所处的二叉树的“层次”。当层次为0时,则表示已经到了叶子节点,递归结束;当层次为1时,则表示已经到了当前要输出数据的层次,在递归调用函数时,将当前节点分别设置为左子树和右子树,并且将当前层次进行减1处理,表示进入二叉树的下一层。
2 递归实现层序遍历的代码
2.1 递归函数的实现
根据图2的流程,实现如下递归函数。
void _bTreeLevelOrder(TreeNode* root, size_t i)
{
if (root == nullptr || i == 0)
{
return;
}
if (i == 1)
{
cout << root->data << endl;
}
_bTreeLevelOrder(root->leftChild, i - 1);
_bTreeLevelOrder(root->rightChild, i - 1);
}
其中,参数root表示当前节点,参数i表示当前二叉树的层次。
2.2 调用递归函数
在定义好递归函数_bTreeLevelOrder()之后,需要再定义一个新的函数来循环调用_bTreeLevelOrder(),代码如下所示。
void bTreeLevelOrder(TreeNode* root)
{
if (root == nullptr)
{
return;
}
int dep = 3;
for (int i = 1; i <= dep; i++)
{
_bTreeLevelOrder(root, i);
}
}
其中,变量dep表示图1所示的二叉树的层次。通过for语句,循环调用_bTreeLevelOrder(),实现二叉树的层序遍历。
最后,在main()函数中调用bTreeLevelOrder()函数即可。