编程之美 - 分层遍历二叉树

问题描述

一棵二叉树，按从上到下，从左到右的方式进行遍历，打印它的所有的节点。

例如二叉树

输出的结果为

a

bc

def

 

思路:

二叉树遍历的方式为分层方式

每一层都可以看做是一个链表，根节点level 0可以看做只有一个元素的链表。当遍历level 0时，可以同时将根节点的左右子树放入level 1的链表中。当遍历level1的链表时可以同理生成level2的链表。完成分层遍历。 level 0，level 1，level 2的链表使用同一个链表即可。

代码示例：

#include 
#include 

using namespace std;

typedef struct _node_st
{
    char _data;
    _node_st* _pleft;
    _node_st* _pright;

} node_st;

list node_list;

void print_tree(node_st* proot)
{
    int count = 1;
    int sub_count = 0;
    int i = 0;
    node_st *pnode = NULL;

    node_list.push_back(proot);
    while(count > 0)
    {
        for (i = 0; i < count; i++)
        {
            pnode = node_list.front();
            node_list.pop_front();
            printf("%c ", pnode->_data);
            if (pnode->_pleft)
            {
                node_list.push_back(pnode->_pleft);
                sub_count++;
            }

            if (pnode->_pright)
            {
                node_list.push_back(pnode->_pright);
                sub_count++;
            }
        }
        printf("\n");
        count = sub_count;
        sub_count = 0;
    }
    printf("\n-----------------------------------------\n");
}

node_st* rebuild_pre_in2(char* pre, char* mid, int start, int length)
{
    node_st *node;
    int j = 0;

    if ((length <= 0) || ((pre == NULL) || (mid == NULL)))
       return NULL;

    node = new node_st;
    node->_data = pre[start];
    node->_pleft = node->_pright = NULL;

    if (1 == length)
        return node;

    for (j = 0; j < length; j++)
    {
        if (mid[start+j] == node->_data)
            break;
    }
    node->_pleft  = rebuild_pre_in2(pre, mid, start+1, j);
    node->_pright = rebuild_pre_in2(pre, mid, start+(j+1), length-j-1);

    return node;
}
void main()
{
    char pre[] = {'a','b','d','c','e','f'};
    char mid[] = {'d','b','a','e','c','f'};
    int len = sizeof(pre)/sizeof(pre[0]);

    node_st* proot = NULL;
    proot = rebuild_pre_in2(pre, mid, 0, len);

    print_tree(proot);
    cin>> len;
}