二叉树遍历(DAY 74)

1：题目

假定一棵二叉树的每个结点都用一个大写字母描述。

给定这棵二叉树的前序遍历和中序遍历，求其后序遍历。

输入格式
输入包含多组测试数据。

每组数据占两行，每行包含一个大写字母构成的字符串，第一行表示二叉树的前序遍历，第二行表示二叉树的中序遍历。

输出格式
每组数据输出一行，一个字符串，表示二叉树的后序遍历。

数据范围
输入字符串的长度均不超过 26。

输入样例：
ABC
BAC
FDXEAG
XDEFAG
输出样例：
BCA
XEDGAF

2：代码实现

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

void dfs(string pre, string in)
{
    if (pre.empty()) return;
    char root = pre[0];
    int k = in.find(root);
    dfs(pre.substr(1, k), in.substr(0, k));
    dfs(pre.substr(k + 1), in.substr(k + 1));
    cout << root;
}

int main()
{
    string pre, in;
    while (cin >> pre >> in)
    {
        dfs(pre, in);
        cout << endl;
    }

    return 0;
}