hihoCoder #1049 : 后序遍历 题解

题目描述

时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述

在参与过了美食节之后，小Hi和小Ho在别的地方又玩耍了一阵子，在这个过程中，小Ho得到了一个非常有意思的玩具——一棵由小球和木棍连接起来的二叉树！小Ho对这棵二叉树爱不释手，于是给它的每一个节点都标记了一个标号——一个属于A..Z的大写字母，并且没有任意两个节点的标号是一样的。小Hi也瞅准了这个机会，重新巩固了一下小Ho关于二叉树遍历的基础知识~就这样，日子安稳的过了两天。这天，小Ho正好在求解这棵二叉树的前序、中序和后序遍历的结果，但是却在求出前序遍历和中序遍历之后不小心把二叉树摔到了地上，小球和木棍等零件散落了一地！小Ho损失了心爱的玩具，正要嚎啕大哭起来，所幸被小Hi发现了，劝说道：“别着急，这不是零件都还在么？拼起来不就是了？”“可是我忘记了二叉树长什么样子了！”小Ho沮丧道。“这个简单，你不是刚刚求出了这棵二叉树的前序和中序遍历的结果么，利用这两个信息就可以还原出整棵二叉树来哦！”这样么？！！”小Ho止住了泪水，问道：“那要怎么做呢？”
没错！小Ho在这一周遇到的问题便是：给出一棵二叉树的前序和中序遍历的结果，还原这棵二叉树并输出其后序遍历的结果。

输入
每个测试点（输入文件）有且仅有一组测试数据。
每组测试数据的第一行为一个由大写英文字母组成的字符串，表示该二叉树的前序遍历的结果。
每组测试数据的第二行为一个由大写英文字母组成的字符串，表示该二叉树的中序遍历的结果。
对于100%的数据，满足二叉树的节点数小于等于26。

输出
对于每组测试数据，输出一个由大写英文字母组成的字符串，表示还原出的二叉树的后序遍历的结果。

样例输入
AB
BA
样例输出
BA

---

题意—-其实题目那么长主要意思就是给定二叉树的前序和中序，输出它的后序。

思路

• 前序的第一个字符一定是根节点，找到之后再在中序中找到它
• 将中序分为左右两棵子树
• 重复以上两步，即可构造出二叉树
• 递归得到二叉树的后序遍历结果

代码写的很丑，其中有用到string中的find()函数以及substr()函数，具体用法参考我的有关这个的博客，之后想对这个代码进行改进，写出更简洁的代码。

代码如下：

#include<iostream>
#include<string>
using namespace std;

struct Node {
    char data;
    Node * left;
    Node * right;
    Node(char a){
        data = a;
        left=NULL;
        right = NULL;
    } 
};

Node* getNode(string front,string center) {
    Node* root = new Node(front[0]);
    int len = center.length();
    if(len == 1){
        return root;
    }
    int index = center.find(front[0]);

    if(index==0  && index !=len-1 ) {
        int temp= len-index-1;
        string subcenter2 = center.substr(index+1,temp);
        string subfront2 = front.substr(1+index,temp);
        root->right = getNode(subfront2,subcenter2);
    } else if(index!=0 && index ==len-1) {
        int temp = index;
        string subcenter1 = center.substr(0,temp);
        string subfront1 = front.substr(1,temp);
        root->left = getNode(subfront1,subcenter1);
    }else{
        int temp1 = index;
        int temp2 = len- index-1;
        string subcenter2 = center.substr(index+1,temp2);
        string subfront2 = front.substr(1+index,temp2);
        string subcenter1 = center.substr(0,temp1);
        string subfront1 = front.substr(1,temp1);
        root->left = getNode(subfront1,subcenter1);
        root->right = getNode(subfront2,subcenter2);
    }
    return root;
}

void printBack(Node *root) {
    if(root->left!=NULL) {
        printBack(root->left);
    }
    if(root->right!=NULL) {
        printBack(root->right);
    }
    cout<<root->data;
    return;
}

int main() {
    string front,center;
    cin>>front>>center;
    Node * tree = getNode(front,center);
    printBack(tree);
    return 0;
}