C++中二叉树的递归遍历方法2-2

在《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()函数即可。