这个题就是转层序,在函数后面加一个index统计各个点的位置就可以了,跟玩转二叉树基本上一样。
代码:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <sstream>
using namespace std;
const int N = 50;
int in[N],post[N],level[100000];
int cnt = 0;
int n;
void dfs(int root, int start, int end, int index){
if(start > end) return;
int i = start;
while(i < end && in[i] != post[root]) i++;
level[index] = post[root];
dfs(root-(end - i + 1),start, i - 1, 2 *index + 1);
dfs(root - 1, i + 1, end, 2 * index + 2);
}
int main(){
scanf("%d",&n);
memset(level, -1, sizeof level);
for(int i=0;i<n;i++) scanf("%d",&post[i]);
for(int i=0;i<n;i++) scanf("%d",&in[i]);
dfs(n-1,0,n-1,0);
for(int i=0;i<10000;i++){
if(level[i] != -1 && cnt != n - 1){
printf("%d ",level[i]);
cnt++;
}else if(level[i] != -1){
printf("%d",level[i]);
break;
}
}
return 0;
}