pat甲级1127题解_c++简洁

题目链接：pat1127A
大意：给出一棵二叉树的中序、后序遍历，要求按z字形输出此二叉树的层次遍历（z字形：根节点为第1层，奇数层从左往右输出，偶数层从右往左输出）
思路：此题难点在于z字形输出，正常层序遍历层数小的优先输出而每层下标小的优先输出，容易想到z字形输出只要在层数为奇数时将下标大的优先输出就可实现。
步骤：在建树过程中保存每个结点所在层数和下标（按完全二叉树设置下标：left_child=parent2，right_child=parent2+1），将所有结点用vector数组保存，最后调用sort对该数组进行排序：层数小的优先，层数相同情况下奇数层下标小优先，偶数层下标大优先。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct node{
    int index;
    int lev;
    int val;
};
vector<int> in,post;
vector<node*> ans;
void build(int b1,int e1,int b2,int e2,int lev,int index){
    if(b1>e1)return;
    int root=post[e2],i=b1;
    for(;i<=e1;i++){
        if(in[i]==root)break;
    }
    node* newn;
    newn=new node;
    newn->lev=lev;
    newn->val=root;
    newn->index=index;
    ans.push_back(newn);
    build(b1,i-1,b2,b2+i-1-b1,lev+1,2*index);
    build(i+1,e1,b2+i-b1,e2-1,lev+1,2*index+1);
}
bool cmp(node* a,node*b){
    if(a->lev!=b->lev)return a->lev<b->lev;
    if(a->lev%2==0)return a->index<b->index;
    return a->index>b->index;
}
int main(){
    int N,val;
    cin>>N;
    for(int i=0;i<N;i++){
        cin>>val;
        in.push_back(val);
    }
    for(int i=0;i<N;i++){
        cin>>val;
        post.push_back(val);
    }
    build(0,in.size()-1,0,post.size()-1,1,1);
    sort(ans.begin(),ans.end(),cmp);
    for(int i=0;i<ans.size()-1;i++){
        cout<<ans[i]->val<<' ';
    }
    cout<<ans.back()->val;
    system("pause");
    return 0;
}