【OJ - 二叉树】二叉树的构建（用一个先序遍历的字符串）及遍历

题目难度： 较难
牛客网链接： 二叉树遍历_

一、题目描述

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

输入描述：

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

输出描述：

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

示例：

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

---

二、解题思路

核心思路：

• 用一个指针依次遍历字符串中每个字符，来递归构建二叉树

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

// 定义链二叉树的节点
typedef struct TreeNode
{
    char val;
    struct TreeNode* left;
    struct TreeNode* right;
}TreeNode;

// 用一串先序遍历的字符串，来构建二叉树，并返回二叉树的根节点地址
TreeNode* CreateBinaryTree(char* str, int* pi)
{
    // 依次遍历字符串中每个字符的值来递归构建二叉树（pi表示字符下标）
    
    // 如果已走到字符串结束标志，则构建完成
    if(str[*pi] == '\0')
    {
        return NULL;
    }
    
    // 1. 如果当前字符为空（“#”表示的是空），构建空树，返回NULL
    if(str[*pi] == '#')
    {
        (*pi)++; // 指向下一个字符的下标
        return NULL;
    }
    
    // 2. 如果当前字符不为空，则动态开辟一个二叉树新节点
    TreeNode* newnode = (TreeNode*)malloc(sizeof(TreeNode));
    if(newnode == NULL)
    {
        perror("malloc");
        exit(-1);
    }
    
    // 3. 初始化新节点，并递归构建二叉树
    newnode->val = str[(*pi)++]; // (*pi)++表示指向下一个字符的下标
    newnode->left = CreateBinaryTree(str, pi);
    newnode->right = CreateBinaryTree(str, pi);
    
    // 4. 返回根节点地址
    return newnode;
}

// 二叉树的中序遍历
void InOrder(TreeNode* root)
{
    if(root)
    {
        // 左、根、右
        InOrder(root->left);
        printf("%c ", root->val); // 每个字符后面都有一个空格
        InOrder(root->right);
    }
}

int main()
{
    char str[100] = { 0 };
    scanf("%s", str); // 输入一行字符串
    
    // 将输入的先序遍历的字符串建立成二叉树（以指针方式存储）
    int i = 0;
    TreeNode* root = CreateBinaryTree(str, &i);
    // 输出二叉树中序遍历结果
    InOrder(root);
    
    return 0;
}

---