地址: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;
}