Sicily 1935. 二叉树重建

1935. 二叉树重建

Constraints

Time Limit: 1 secs, Memory Limit: 32 MB

Description

对于二叉树T，可以递归定义它的先序遍历、中序遍历和后序遍历如下： PreOrder(T)=T的根节点+PreOrder(T的左子树)+PreOrder(T的右子树) InOrder(T)=InOrder(T的左子树)+T的根节点+InOrder(T的右子树) PostOrder(T)=PostOrder(T的左子树)+PostOrder(T的右子树)+T的根节点 其中加号表示字符串连接运算。例如，对下图所示的二叉树，先序遍历为DBACEGF，中序遍历为ABCDEFG。

输入一棵二叉树的先序遍历序列和中序遍历序列，输出它的广度优先遍历序列。

Input

第一行为一个整数t（0<t<10），表示测试用例个数。 以下t行，每行输入一个测试用例，包含两个字符序列s1和s2，其中s1为一棵二叉树的先序遍历序列，s2为中序遍历序列。s1和s2之间用一个空格分隔。序列只包含大写字母，并且每个字母最多只会出现一次。

Output

为每个测试用例单独一行输出广度优先遍历序列。

Sample Input

2 
DBACEGF ABCDEFG 
BCAD CBAD

Sample Output

DBEACGF 
BCAD
// Problem#: 1935
// Submission#: 3359425
// The source code is licensed under Creative Commons Attribution-NonCommercial-ShareAlike 3.0 Unported License
// URI: http://creativecommons.org/licenses/by-nc-sa/3.0/ // All Copyright reserved by Informatic Lab of Sun Yat-sen University
#include <iostream>
#include <vector>
using namespace std;

struct TreeNode {
    char v;
    TreeNode * l, * r;
    TreeNode(char val = 0, TreeNode * ll = NULL, TreeNode * rr = NULL) {
        v = val;
        l = ll;
        r = rr;
    }
};

string pre, in;
vector<char> ans[30];  // 存放广度优先遍历的序列

TreeNode * rebuild(int preS, int preE, int inS, int inE) {  // preS、preE分别是当前前序遍历序列的起始和末尾，inS、inE分别是当前中序遍历序列的起始和末尾
    if (preS > preE || inS > inE) return NULL;  // 如果树是空
    TreeNode * now = new TreeNode(pre[preS]);
    int pos;
    for (pos = inS; pos <= inE; pos++) {  // 在中序遍历中找到当前跟节点
        if (pre[preS] == in[pos]) break;
    }
    now->l = rebuild(preS + 1, preS + pos - inS, inS, pos - 1);  // 递归重建左子树，注意左子树序列长度是pos-inS
    now->r = rebuild(preS + pos - inS + 1, preE, pos + 1, inE);  // 递归重建右子树
    return now;
}

void preOrder(TreeNode * now, int floor) {  // 得到广度优先遍历序列
    if (now->l) preOrder(now->l, floor + 1);
    ans[floor].push_back(now->v);
    if (now->r) preOrder(now->r, floor + 1);
}

int main() {

    int caseNum;
    cin >> caseNum;

    while (caseNum--) {
        cin >> pre >> in;
        TreeNode * root = rebuild(0, pre.size() - 1, 0, in.size() - 1);
        preOrder(root, 0);
        for (int i = 0; ans[i].size() != 0; i++) {  // 输出广度优先遍历序列
            for (int j = 0; j < ans[i].size(); j++) cout << ans[i][j];
            ans[i].clear();
        }
        cout << endl;
    }

    return 0;
}