解题思路:
1.根据中序和后序建立二叉树
2.按照zigzag层序输出
#include<cstdio>
#include<vector>
#include<queue>
#include<iostream>
using namespace std;
int n;
vector<int> in,post;
vector<int> zig[35];
struct node{
int data,layer;
node *left, *right;
node(int x):data(x),left(NULL),right(NULL){}
};
void create(node* &root, int postL,int postR, int inL, int inR){
if(postL > postR) return;
if(root == NULL)
root = new node(post[postR]); //这个地方容易写错
int k=inL;
while(root->data != in[k]){
k++;
}
int left = k - inL;
create(root->left,postL, postL+left-1,inL, k-1);
create(root->right,postL+left,postR-1, k+1, inR);
}
void zigzagorder(node* root){
queue<node*> q;
q.push(root);
root->layer = 0;
while(!q.empty()){
node* temp = q.front();
q.pop();
zig[temp->layer].push_back(temp->data);
if(temp->right!=NULL) temp->right->layer=temp->layer+1;
if(temp->left!=NULL) temp->left->layer=temp->layer+1;
if(temp->left!=NULL) q.push(temp->left);
if(temp->right!=NULL) q.push(temp->right);
}
}
int main(){
int x;
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&x);
in.push_back(x);
}
for(int i=0;i<n;i++){
scanf("%d",&x);
post.push_back(x);
}
node* root = NULL;
create(root,0,post.size()-1,0,in.size()-1);
zigzagorder(root);
int cnt=0;
for(int i=0;i<n;i++){
if(i&1){
for(int j=0;j<zig[i].size();j++){
cnt++;
if(cnt != n) printf("%d ",zig[i][j]);
else {
printf("%d\n",zig[i][j]);
break;
}
}
}else{
for(int j=zig[i].size()-1;j>=0;j--){
cnt++;
if(cnt != n) printf("%d ",zig[i][j]);
else {
printf("%d\n",zig[i][j]);
break;
}
}
}
}
return 0;
}