#include<iostream>
#include<vector>
using namespace std;
vector<int> pre,pos,in;
struct Node{
int data;
Node* left;
Node* right;
};
int flag=1;
int n;
Node* build(int prel,int prer,int posl,int posr){
if(prel>prer) return NULL;
Node* root=new Node;
root->data=pre[prel];
root->left=NULL;
root->right=NULL;
if(posl==posr) return root;
int k;
for(k=posl;k<=posr;k++){
if(pre[prel+1]==pos[k]){
break;
}
}
if(k==posr-1){
flag=0;
}
int numleft=k-posl+1;
root->left=build(prel+1,prel+numleft,posl,k);
root->right=build(prel+numleft+1,prer,k+1,posr-1);
return root;
}
void intra(Node* root){
if(root==NULL) return;
intra(root->left);
in.push_back(root->data);
intra(root->right);
}
int main(){
#ifdef ONLINE_JUDGE
#else
freopen("in.txt","r",stdin);
#endif
cin>>n;
pre.resize(n);
pos.resize(n);
for(int i=0;i<n;i++){
cin>>pre[i];
}
for(int i=0;i<n;i++){
cin>>pos[i];
}
Node* root=build(0,n-1,0,n-1);
intra(root);
if(flag==0) cout<<"No"<<endl;
else cout<<"Yes"<<endl;
for(int i=0;i<in.size();i++){
if(i==0) cout<<in[i];
else cout<<' '<<in[i];
}
return 0;
}
2个问题,build函数里面加了一个prel==prer的时候返回root的语句,为什么之前写前序中序或者后序中序找层序的时候建树不需要这个判断语句?而且貌似写上posl==posr这个判断语句是等价的?prel>prer时返回NULL如何理解?
这里建树的时候的k标志着左子树的根,而根据中序和前序建树的时候的k标志着整体的根,所以在这里要加上判断语句,即判断posl==posr,这个时候返回root即可,这个root就是只含一个节点的左子树,实际上画个图就能理解了