二叉树遍历——清华大学复试上机

二叉树遍历——清华大学复试上机

题目描述

题目描述
编一个程序，读入用户输入的一串先序遍历字符串，根据此字符串建立一个二叉树（以指针方式存储）。 例如如下的先序遍历字符串： ABC##DE#G##F### 其中“#”表示的是空格，空格字符代表空树。建立起此二叉树以后，再对二叉树进行中序遍历，输出遍历结果。
输入描述:
输入包括1行字符串，长度不超过100。
输出描述:
可能有多组测试数据，对于每组数据，
输出将输入字符串建立二叉树后中序遍历的序列，每个字符后面都有一个空格。
每个输出结果占一行。
示例1
输入
abc##de#g##f###
输出
c b e g d f a

进行对题目的解析

  题目已知中输入先序遍历的结果，求中序遍历。
  从先序遍历入手，根据先序遍历的特性（根左右）复刻出完整的树。
先序遍历：ABC##DE#G##F###
复刻出完整的树。

如果看到此处不懂如何复刻出的这颗树，可以继续向下看，下面有配图详解。

创建结构体

typedef struct TreeNode {
    char val;             //数据域
    struct TreeNode* left; //左孩子
    struct TreeNode* right; //有孩子
}TreeNode;

核心代码

//str表示输入的字符串
//pi表示指向字符串的指针
TreeNode* preOrder(char* str, int& pi)
{
    if (str[pi] == '#')
    {
        ++pi;
        return NULL;
    }
    //创建一个结点存储数据。
    TreeNode* root = (TreeNode*)malloc(sizeof(TreeNode));
    if (root == NULL)
    {
        cout << "malloc is fail" << endl;
        exit(-1);
    }
    root->val = str[pi];
    ++pi;
    root->left = preOrder(str, pi);
    root->right = preOrder(str, pi);

    return root;

}

配图详解

  当指针pi指向的位置是a，不为#,创建一个结点存储数据A。将A看作一个根结点,开始遍历A结点的左子树。

  当指针pi指向的位置是b，不为#,创建一个结点存储数据B。将B看作一个根结点,开始遍历B结点的左子树。

  当指针pi指向的位置是C，不为#,创建一个结点存储数据C。将C看作一个根结点,开始遍历C结点的左子树。

  当指针pi指向的位置是**#，说明C的左子树是空。开始遍历C的右子树。

  当指针pi指向的位置是#，说明C的右子树是空。返回这个递归，开始遍历B的右子树。

  当指针pi指向的位置是d**，不为#,创建一个结点存储数据D。将D看作一个根结点,开始遍历D结点的左子树。

  当指针pi指向的位置是e，不为#,创建一个结点存储数据E。将E看作一个根结点,开始遍历E结点的左子树。

  当指针pi指向的位置是**#，说明E的左子树是空。返回这个递归，开始遍历E的右子树。

  当指针pi指向的位置是g**，不为#,创建一个结点存储数据G。将G看作一个根结点,开始遍历G结点的左子树。

  当指针pi指向的位置是**#，说明G的左子树是空。返回这个递归，开始遍历G**的右子树。

  当指针pi指向的位置是**#，说明G的右子树是空。返回这个递归，开始遍历D**的右子树。

  当指针pi指向的位置是f，不为#,创建一个结点存储数据F。将F看作一个根结点,开始遍历F结点的左子树。

  当指针pi指向的位置是**#，说明F的左子树是空。返回这个递归，开始遍历F的右子树。

  当指针pi指向的位置是#，说明F的右子树是空。返回这个递归，开始遍历A的右子树。

  当指针pi指向的位置是#，说明A**的右子树是空。全部结点遍历完毕，结束。

要将得到的树中序遍历

中序遍历核心代码

void InOrder(TreeNode* root)
{
    if (root == NULL)
    {
        return;
    }
    InOrder(root->left);
    cout << root->val<<" ";
    InOrder(root->right);
}

源代码

#include<iostream>
using namespace std;

typedef struct TreeNode {
    char val;             //数据域
    struct TreeNode* left; //左孩子
    struct TreeNode* right; //有孩子
}TreeNode;

//str表示输入的字符串
//pi表示指向字符串的指针
TreeNode* preOrder(char* str, int& pi)
{
    if (str[pi] == '#')
    {
        ++pi;
        return NULL;
    }
    //创建一个结点存储数据。
    TreeNode* root = (TreeNode*)malloc(sizeof(TreeNode));
    if (root == NULL)
    {
        cout << "malloc is fail" << endl;
        exit(-1);
    }
    root->val = str[pi];
    ++pi;
    root->left = preOrder(str, pi);
    root->right = preOrder(str, pi);

    return root;

}
void InOrder(TreeNode* root)
{
    if (root == NULL)
    {
        return;
    }
    InOrder(root->left);
    cout << root->val<<" ";
    InOrder(root->right);
}
int main()
{
    char str[105];
    cin >> str;
    int i = 0;
    TreeNode* root = preOrder(str, i);
    InOrder(root);
    return 0;
}

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

运行结果

觉得我回答有用的话，记得点个关注哟！谢谢支持！