【二叉树练习】 - 还原二叉树

题目描述

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

示例1

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

如：

输入：abc##de#g##f###
输出：c b e g d f a

---

解题分析：

思路：
注意：一般情况下，前序+中序或后序+中序，可以确定唯一的一颗二叉树。如果前序或中序或后序有空树表示，也能确定二叉树。
先确定根节点，再确定左子树的根，直到子树节点为空，返回。再确定右子树的根，直到子树节点为空。
当根，左子树，右子树都确定了，还原一颗子树。

---

代码实现：

#include <stdio.h>
#include <stdlib.h>
typedef struct TreeNode
{
    
    
	struct TreeNode* left;//节点的左子树
	struct TreeNode* right;//节点的右子树
	char data;//节点的值
}TNode;

//思想：类似前序遍历
TNode* CreateTree(char* a,int* pi)
{
    
    
	if (a[*pi] == '#')//空树
	{
    
    
		++(*pi);//这一行必须要，下一个字符
		return NULL;
	}

	//构建树
	TNode* root = (TNode*)malloc(sizeof(TNode));//每个树的根节点
	if (root == NULL)
	{
    
    
		printf("malloc failed\n");
		exit(-1);
	}
	//当前节点（根）
	root->data = a[*pi];
	++(*pi);
	//当前节点的左子树
	root->left = CreateTree(a,pi);
	//当前节点的右子树
	root->right = CreateTree(a,pi);

	return root;
}

//中序遍历
void InOrder(TNode* root)
{
    
    
	if (root == NULL)
		return;
	InOrder(root->left);
	printf("%c ", root->data);
	InOrder(root->right);
}

int main()
{
    
    
	//测试功能
	char str[100];//由于告诉了字符串不会超过100个字节，可以直接开辟固定大小的数组。
	scanf("%s", str);
	int i = 0;//字符串的下标
	TNode* root = CreateTree(str, &i);//由于下一层的i不会影响上一层的i，i需要取地址

	InOrder(root);
	return 0;
}

---