二叉树的创建和文本显示

二叉树的创建和文本显示

时间限制: 1 Sec
内存限制: 128 MB

题目描述

编一个程序，读入先序遍历字符串，根据此字符串建立一棵二叉树（以指针方式存储）。

例如如下的先序遍历字符串： A ST C # # D 10 # G # # F # # #

各结点数据（长度不超过3），用空格分开，其中“#”代表空树。
建立起此二叉树以后，再按要求输出二叉树。

输入

输入由多组测试数据组成。

每组数据包含一行字符串，即二叉树的先序遍历，字符串长度大于0且不超过100。

输出

对于每组数据，显示对应的二叉树，然后再输出一空行。输出形式相当于常规树形左旋90度。见样例。 注意二叉树的每一层缩进为4，每一行行尾没有空格符号。

样例输入

A ST C # # D 10 # G # # F # # #
4 2 1 # # 3 # # 5 # 6 # #

样例输出

A
            F
        D
                G
            10
    ST
        C

        6
    5
4
        3
    2
        1

解题思路

根据输出顺序可以观察出是按照中序遍历输出但是是按照镜像输出方式（先输出右子树再输出左子树）进行输出的。本题难点在于求树的深度及每一层的层数，根据层数输出前面的缩进。

AC代码

#include <bits/stdc++.h>
using namespace std;

typedef struct T
{
    char data[4];
    struct T *rchild,*lchild;
} TNode,*Tree;

char a[100][3]; //模拟树的每层的字符
int depth = 0;

void ini_T(Tree &T)
{
    T = new TNode;
    T->lchild = NULL;
    T->rchild = NULL;
}

void Create_T(Tree &t)
{
    if(a[depth][0] == '#')
    {
        t = NULL;
        depth++;  //当一个结点为空时意味着该结点的父节点创建成功，深度+1
    }
    else
    {
        t = new TNode;
        strcpy(t->data,a[depth++]);
        Create_T(t->lchild);
        Create_T(t->rchild);
    }
}

void MINTrave(Tree t,int dep)  //镜像中序遍历二叉树
{
    if(t)
    {
        MINTrave(t->rchild,dep+1); //先遍历右子树

        int n = dep * 4 ;
        for(int i = 0; i < n; ++i) //每层先输出4倍层数的空格
            cout << ' ';

        cout << t->data << endl;

        MINTrave(t->lchild,dep+1);
    }
}

int main()
{
    char b[1000];
    while(gets(b) != NULL)
    {
        Tree T;
        ini_T(T);
        int i, p = 0,q = 0;
        for(i = 0 ; b[i] ; ++i)
        {
            if(b[i] != ' ')
                a[p][q++] = b[i];
            else      //遇到空格时，该结点创建完毕
            {
                a[p++][q] = '\0'; //p++ 移到下一个结点
                q = 0;
            }
        }
        a[p++][q++] = '\0';
        Create_T( T );
        MINTrave( T, 0 );
        depth = 0;  //将深度更新为0
        cout<<endl;
    }
    return 0;
}