树的重建

#include<iostream>
#include<string>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> pre,in,post;
int n,pos;
void rec(int l,int r) {
if (l>=r) return ;
int root=pre[pos++];                                                                                    //确定结点
int m=distance(in.begin(), find(in.begin(),in.end(),root));                                 //早到结点在in的位置  确定子节点位置

rec(l,m); //左节点递归

rec(m+1,r);                                                                                                     //  右节点递归
post.push_back(root);                           //压人post栈
}
void selove() {
pos=0;
rec(0,pre.size());                                    //pre 遍历结果确定了结点顺序
for (int i=0;i<n;i++) {
if (i) cout<<' ';
cout<<post[i];
}
cout << endl;
}
int main() {
int k;
cin>>n;
for (int i=0;i<n;i++) {
cin>>k;
pre.push_back(k);
}
for (int i=0;i<n;i++) {
cin>>k;
in.push_back(k);
}
selove();
return 0;
}

猜你喜欢

转载自blog.csdn.net/acm160920007/article/details/76202496