二叉树遍历 题解（牛客网）

描述

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

输入描述：

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

输出描述：

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

示例1

输入：

abc##de#g##f###

复制输出：

c b e g d f a 

题解

输入的那串字符串是先序遍历的，要按照先序遍历的方式将字符串中的内容建立二叉树存储，如图

1.定义二叉树节点结构

typedef int BTDataType;
typedef struct BinaryTreeNode
{
	BTDataType data;
	struct BinaryTreeNode* left;
	struct BinaryTreeNode* right;
}BTNode;

BTNode* BuyNode(BTDataType x)
{
	BTNode* node = (BTNode*)malloc(sizeof(BTNode));
	if (node == NULL)
	{
		perror("malloc fail");
		return NULL;
	}

	node->data = x;
	node->left = NULL;
	node->right = NULL;

	return node;
}

2.建立二叉树

我们使用一个全局指针* pi 来追踪当前处理到先序字符串的位置：

• 若当前字符是 '#'，表示空节点，*pi 后移，返回 NULL。
• 若当前字符是有效节点，创建节点，*pi 后移，递归构建左子树，再递归构建右子树。

BTNode* CreateTree(char*a,int* pi)
{
    if(a[*pi] == '#')
    {
        (*pi)++;
        return NULL;
    }

    BTNode* root = BuyNode(a[*pi]);
    (*pi)++;
    root->left = CreateTree(a,pi);
    root->right = CreateTree(a,pi);
    return root;
}

3，中序遍历

中序遍历的顺序是 左子树 → 根节点 → 右子树，同样采用递归实现：

void InOrder(BTNode* root)
{
    if(root == NULL)
    {
        return;
    }
    InOrder(root->left);
    printf("%c ",root->data);
    InOrder(root->right);
}

完整代码

#include <stdio.h>
#include<stdlib.h>

typedef int BTDataType;
typedef struct BinaryTreeNode
{
	BTDataType data;
	struct BinaryTreeNode* left;
	struct BinaryTreeNode* right;
}BTNode;

BTNode* BuyNode(BTDataType x)
{
	BTNode* node = (BTNode*)malloc(sizeof(BTNode));
	if (node == NULL)
	{
		perror("malloc fail");
		return NULL;
	}

	node->data = x;
	node->left = NULL;
	node->right = NULL;

	return node;
}

BTNode* CreateTree(char*a,int* pi)
{
    if(a[*pi] == '#')
    {
        (*pi)++;
        return NULL;
    }

    BTNode* root = BuyNode(a[*pi]);
    (*pi)++;
    root->left = CreateTree(a,pi);
    root->right = CreateTree(a,pi);
    return root;
}

void InOrder(BTNode* root)
{
    if(root == NULL)
    {
        return;
    }
    InOrder(root->left);
    printf("%c ",root->data);
    InOrder(root->right);
}

int main() {
    char a[100];
    scanf("%s",&a);
    int i = 0;
    BTNode* root = CreateTree(a,&i);
    InOrder(root);
    printf("\n");
    return 0;
}