地址:http://acm.hdu.edu.cn/showproblem.php?pid=1710
思路:通过对先序遍历和中序遍历的遍历顺序分析,对于先序遍历,第一个节点一定是根节点,再定位到中序遍历中当前根节点的位置,可以将先序遍历拆分成两个子树的先序遍历,这样就可以利用递归将遍历集合一步步缩小,从而得到解。
Code:
#include<iostream>
#include<unordered_map>
using namespace std;
const int MAX_N=1005;
int n,tt;
int pre[MAX_N],in[MAX_N];
unordered_map<int,int> imap;
void DFS(int l,int r,int l1);
int main()
{
ios::sync_with_stdio(false);
while(cin>>n){
imap.clear();
for(int i=0;i<n;++i)
cin>>pre[i];
for(int i=0;i<n;++i)
{
cin>>in[i];
imap[in[i]]=i;
}
tt=0;
DFS(0,n-1,0);
cout<<endl;
}
return 0;
}
void DFS(int l,int r,int l1)
{
if(l>r) return;
int t=imap[pre[l]];
DFS(l+1,l+t-l1,l1);
DFS(l+t-l1+1,r,t+1);
if(tt) cout<<" ";
cout<<pre[l];
tt=1;
}