二叉树的创建和文本显示

题目描述

编一个程序，读入先序遍历字符串，根据此字符串建立一棵二叉树（以指针方式存储）。
例如如下的先序遍历字符串：
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

考点：二叉树的建立，遍历，和输出格式

思路：

1.根据先序输入的字符串转换为可利用的数据

2.把数据用先序遍历存储在二叉树中

3.后序遍历一遍二叉树，然后存储在输出数组中（我用pair封装，方便输出）

4统计有效节点的个数，然后根据每个节点的深度输出相应的空格

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <utility>

using namespace std;

typedef struct node
{
    char s[5];
    int depth;
    struct node *lchild,*rchild;
} tree_node,*xtree;
typedef pair <char [100],int> pa;
char data[100][100];
int data_num;//数据的个数
int data_index;//数据索引
pa ans[100];//存储答案
void creat_xtree(xtree &t,int depth)
{
    if(data[data_index][0]=='#')
    {
        t=NULL;
        data_index++;
    }
    else
    {
        t = new tree_node;
        strcpy(t->s,data[data_index++]);
        t->depth = depth;
        creat_xtree(t->lchild,depth+1);
        creat_xtree(t->rchild,depth+1);
    }
}

void over (xtree t)
{
    if(t)
    {
        over(t->rchild);
        strcpy(ans[data_index].first,t->s);
        //printf("s=%s  ",t->s);
        ans[data_index].second = t->depth;
        //printf("depth=%d\n",t->depth);
        data_index++;
        over(t->lchild);
    }
}
void output()
{
    int k,kk=0;
    for(k=0;k<=data_num;k++)
        if(strcmp(data[k],"#")==0) kk++;
    //统计输出的高度
    for(int i=0;i<=data_num-kk;i++)
    {
        for(int j=0;j<ans[i].second;j++)
            printf("    ");
        printf("%s\n",ans[i].first);
    }
    printf("\n");
}
int main()
{
    char s[200];
    xtree TREE;
    while(gets(s)!=NULL)
    {
        int k=0;
        data_num = data_index = 0;
        for(int i=0; i<strlen(s); i++)
        {
            if(s[i]==' ')
            {
                data[data_num][k++]='\0';
                data_num++;
                k=0;
            }
            else
                data[data_num][k++]=s[i];
        }
        data[data_num][k]='\0';
        //输入数据转化为二维数组
//            for(int i=0;i<=data_num;i++)
//                puts(data[i]);

        creat_xtree(TREE,0);//创建一棵二叉树
        data_index=0;
        over(TREE);//后序遍历检测一下,把输出格式化
        output();
    }
    return 0;
}