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;
}