算法笔记-问题 D: 二叉树遍历

问题 D: 二叉树遍历

题目描述

编一个程序，读入用户输入的一串先序遍历字符串，根据此字符串建立一个二叉树（以指针方式存储）。
例如如下的先序遍历字符串：
ABC##DE#G##F###
其中“#”表示的是空格，空格字符代表空树。建立起此二叉树以后，再对二叉树进行中序遍历，输出遍历结果。

输入

 

输入包括1行字符串，长度不超过100。

 

输出

 

可能有多组测试数据，对于每组数据，
输出将输入字符串建立二叉树后中序遍历的序列，每个字符后面都有一个空格。
每个输出结果占一行。

 

样例输入 Copy

a#b#cdef#####
a##

样例输出 Copy

a b f e d c 
a 

代码：

#include <bits/stdc++.h>

using namespace std;
const int maxn = 50;
int i;
struct node{
    char data;
    node* lchild;
    node* rchild;
}BiTree;
char pre[maxn];

/*重建*/
node* restore(){
    if(pre[i]=='#'){
        i++;
        return NULL;
    }
    node* root = new node;
    root->data = pre[i];
    i++;
    root->lchild = restore();
    root->rchild = restore();
    return root;
}

/*中序*/
void inorder(node* root){
    if(root == NULL){
        return;
    }
    inorder(root->lchild);
    printf("%c ", root->data);
    inorder(root->rchild);
}

int main(){
    while(scanf("%s", pre)!=EOF){
        i = 0;
        node* root = new node;
        root = restore();
        inorder(root);
        printf("\n");
        memset(pre, ' ', sizeof(pre));
    }
	return 0;
}