pat a 1127

这道题的输出让我写了好久，我一开始一直在想在层序遍历中找到某一行的第一个和最后一个，但是很不直观，我老是出错。后来换成把某一层单独存放，然后根据层数是奇数还是偶数来正着输出还是倒着输出，这样就好输出多了。

#include <iostream>
#include <vector>
#include <queue>
using namespace std;

struct node{
	int data;
	int layer;
	node* left;
	node* right;
};

vector<int> post,in;
vector<node*> level;
vector<int> l[35];
vector<int> zig;

node* build(int inl,int inr,int pol,int por){
	if(inl>inr){
		return NULL;
	}
	node* root=new node;
	root->data=post[por];
	root->left=NULL;
	root->right=NULL;
	if(inl==inr){
		return root;
	}
	int k;
	for(int i=inl;i<=inr;i++){
		if(post[por]==in[i]){
			k=i;
			break;
		}
	}
	root->left=build(inl,k-1,pol,pol+k-inl-1);
	root->right=build(k+1,inr,pol+k-inl,por-1);
	return root;
}

void levelorder(node* root,int lay){
	queue<node*> q;
	root->layer=lay;
	q.push(root);
	while(!q.empty()){
		node* top=q.front();
		level.push_back(top);
		q.pop();
		if(top->left!=NULL){
			top->left->layer=top->layer+1;
			q.push(top->left);
		}
		if(top->right!=NULL){
			top->right->layer=top->layer+1;
			q.push(top->right);
		}
	}	
}

void zigzag(){
	int maxl=level[level.size()-1]->layer;
	for(int i=0;i<level.size();i++){
		l[level[i]->layer].push_back(level[i]->data);
	}
	
	for(int i=1;i<=maxl;i++){
		if(i%2==1){
			for(int j=l[i].size()-1;j>=0;j--){
				zig.push_back(l[i][j]);
			}
		}else{
			for(int j=0;j<l[i].size();j++){
				zig.push_back(l[i][j]);
			}
		}
	}
	
}


int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		int temp;
		cin>>temp;
		in.push_back(temp);
	}
	
	for(int i=0;i<n;i++){
		int temp;
		cin>>temp;
		post.push_back(temp);
	} 
	
	node* root=NULL;
	root=build(0,n-1,0,n-1);
	levelorder(root,1);
	zigzag();
	
	for(int i=0;i<zig.size();i++){
		printf("%d",zig[i]);
		if(i!=zig.size()-1){
			printf(" ");
		}
	}
	
	return 0;
}