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

题意解析：已知前中序遍历重构原树，通过观察，前序的第一个结点就是整棵树的根结点，在中序中找到此元素的位置i，0~i-1为左子树，i~n-1(n是pre的长度）为右子树。同理前序第二个结点可以继续将左子树划分为左右子树两个区间……这样下去，当左边不再存在区间时，返回上一级，对另一半的区间进行划分，重复上述步骤，直到全部划分完毕。用递归实现，之后用队列实现BFS，输出广度优先遍历序列。

代码如下：

#include<iostream>#include<queue>#include<string> using namespace std;string pre,in;int index,len;struct Tree{       Tree *left;       Tree *right;       char data;       Tree(char ch=0)       {           left=right=NULL;           data=ch;       }};void rebuild(Tree *&root,int st,int ed){     if(st>ed||index>=pre.size())return ;          root = new Tree(pre[index]);     int root_index=in.find(pre[index++]);          rebuild(root->left,st,root_index-1);     rebuild(root->right,root_index+1,ed); } int main(){    int t;    cin>>t;    while(t--)    {        cin>>pre>>in;        len=pre.size();        index=0;        Tree *root=new Tree();        rebuild(root,0,len-1);                queue<Tree*>q;        q.push(root);        while(!q.empty())        {            Tree* tmp=q.front();            q.pop();            cout<<tmp->data;            if(tmp->left)q.push(tmp->left);            if(tmp->right)q.push(tmp->right);                    }        cout<<endl;    }    return 0;}