给定一棵二叉树的先序遍历和中序遍历,请你输出其后序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(≤30),是二叉树中结点的个数。第二行给出其先序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的后序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
6 3 5 4 2 6 1 4 5 2 3 1 6
输出样例:
4 2 5 1 6 3
#include<bits/stdc++.h>
using namespace std;
const int N=10010;
int per[N],in[N],post[N];
int k;
struct node{
int val;
node *left,*right;
node(int val=0,node *left=NULL,node *right=NULL):val(val),left(left),right(right){}
};
void buildtree(int l,int r,int &t,node *&root){//建树
int flag=-1;
for(int i=l;i<=r;i++){
if((in[i]==per[t])) {
flag=i;break;
}
}
if(flag==-1) return;
root=new node(in[flag]);
t++;
if(flag>l) buildtree(l,flag-1,t,root->left);
if(flag<r) buildtree(flag+1,r,t,root->right);
}
//后序遍历
void hou(node *root){
if(root!=NULL){
hou(root->left);
hou(root->right);
post[k++]=root->val;
}
}
void re_free(node *root){
if(root==NULL) return ;
re_free(root->left);
re_free(root->right);
delete root;
}
int main(){
int n;
while(~scanf("%d",&n)){
for(int i=1;i<=n;i++) scanf("%d",&per[i]);
for(int i=1;i<=n;i++) scanf("%d",&in[i]);
node *root;
int t=1;
buildtree(1,n,t,root);
k=0;
hou(root);
for(int i=0;i<k;i++){
printf("%d",post[i]);
if(i!=k-1)
printf(" ");
}
re_free(root);
}
return 0;
}