题目大意:前序中序求后序
老套路题了,不过需要注意N最大取到50000,正常会超时,但是数据好像没取到极限,正常做刚好可以过。
为什么会超时,正常做的时候会耗费时间去查找根结点,所以我们直接用哈希表优化。
可以试试用哈希表和不用,确实速度差了好多。
#include <iostream>
#include <unordered_map>
using namespace std;
const int N = 50010;
int n;
int pre[N],in[N],f = 0;
unordered_map<int,int>mp;
void dfs(int preL,int preR,int inL,int inR)
{
if(preL > preR || f) return ;
//int k = inL;
//while(pre[preL] != in[k]) k ++;
int k = mp[pre[preL]];//哈希表优化
dfs(preL + 1,preL + k - inL,inL,k -1);
dfs(preL + k - inL + 1,preR,k + 1,inR);
if(f == 0)
cout << pre[preL];f = 1;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> n;
for(int i = 0; i < n; i ++) cin >> pre[i];
for(int i = 0; i < n; i ++)
{
cin >> in[i];
mp[in[i]] = i;
}
dfs(0,n-1,0,n-1);
return 0;
}