1127 ZigZagging on a Tree（30 分）【树的遍历】

题意：已知中序遍历和后序遍历，建树，输出层序遍历，输出规则为：从左到右，再从右到左，交替输出

#include <bits/stdc++.h>
using namespace std;
#define INF 0x7FFFFF

int n;
int in[31],po[31];

typedef struct node
{
    int val;
    struct node *lchild,*rchild;
}node;

node *buildT(int is,int ie,int ps,int pe)
{
    if(is>ie)
        return NULL;
    int t=0;
    while(in[t]!=po[pe])
        t++;
    node *binode=(node*)malloc(sizeof(node));
    binode->val=po[pe];
    binode->lchild=buildT(is,t-1,ps,ps+t-is-1);
    binode->rchild=buildT(t+1,ie,ps+t-is,pe-1);
    return binode;
}

void bfs(node *root)
{
    queue<node*> q;
    q.push(root);
    cout<<root->val;

    int level=2;  //数当前的层数，从第2层开始
    while(!q.empty()){
        vector<node*> v;  //用vector存储每一层的结点
        while(!q.empty()){
            node *temp=NULL;
            temp=q.front();
            q.pop();
            if(temp->lchild)
                v.push_back(temp->lchild);
            if(temp->rchild)
                v.push_back(temp->rchild);
        }
        for(int i=0;i<v.size();i++)
            q.push(v[i]);
        if(level%2==0){  //当层数为偶数时，从左往右输出
            for(int i=0;i<v.size();i++)
                cout<<" "<<v[i]->val;
        }
        else{  //层数为奇数时，从右往左输出
            for(int i=v.size()-1;i>=0;i--)
                cout<<" "<<v[i]->val;
        }
        level++;
    }
    cout<<endl;
}

int main()
{
    cin>>n;
    for(int i=0;i<n;i++){
        cin>>in[i];
    }
    for(int i=0;i<n;i++){
        cin>>po[i];
    }
    node *root;
    root=buildT(0,n-1,0,n-1);
    bfs(root);

    return 0;
}