二叉树的分层打印（二）

在上一篇文章“二叉树的分层打印（一）”中，已经实现了二叉树的分层打印，本文讲述了其他的实现方式。

思路分析：
<1> 如果只是分层遍历二叉树并打印出所有元素，那么使用队列来实现BFS是最好的选择。
<2> 但是这里要求，每层元素打印为一行，所以我们必须知道每层元素的开始和结束是什么，这种情况下，使用数组或者STL中的vector容器是更好的选择，使用两个变量来标识每一层的开始和结束，控制每一层元素的访问。 这里我们使用vector来代替数组，可利用其动态扩展的属性。具体实现如下：

#include <iostream>
using namespace std;
#include <vector>


typedef struct node
{
    char data;
    struct node *lchild;
    struct node *rchild;
}BiNode, *BiTree;


// 先序建立二叉树 （输入时，按先序次序输入二叉树中结点的值,以 # 字符表示空树）
BiTree createBiTree()
{
    BiTree T;

    char c;
    scanf("%c", &c);

    if (c == '#')
        T = NULL;
    else
    {
        T = new BiNode;  // 或 T = (BiTree)malloc(sizeof(BiNode));
        T->data = c;

        T->lchild = createBiTree();
        T->rchild = createBiTree();
    }

    return T;

}


// 分层打印二叉树
void PrintTreeByLevel(BiTree T)
{
    if (NULL == T) return;

    vector<BiTree> vec;
    vec.push_back(T);

    int cur = 0; // 指向当前节点
    int last = 1; // 当前vec中节点的个数

    while(cur < vec.size())
    {
        last = (int)vec.size(); // 新的一层访问开始，last位于当前层最后一个节点的下一个位置(注意：last表示当前vec中节点的“个数”，是个数，不是下标)
        while (cur < last)
        {
            printf("%c ", vec[cur]->data);
            if (vec[cur]->lchild)
                vec.push_back(vec[cur]->lchild);

            if (vec[cur]->rchild)
                vec.push_back(vec[cur]->rchild);

            cur++;
        }
        //每一次循环都将下一层的节点压入vector，并且打印完本层节点。

        printf("\n"); // 当cur == last时,说明该层访问结束，输出换行符
    }
}


int main(int argc, const char * argv[]) {

    BiTree T = createBiTree(); // 建立

     PrintTreeByLevel(T); // 分层打印二叉树

    return 0;
}