天梯赛 L2-006 树的遍历（序列建树）

#include<bits/stdc++.h>
#define MAX 125
using namespace std;

struct Node{   //二叉树建立
    int x,l,r;
}tree[MAX];
int h[MAX],z[MAX];
int c;

int find(int h[],int z[],int len){   //找中序的父节点 
    for(int i=0;i<len;i++){
        if(z[i]==h[len-1]){
            return i;
        }
    }
}
int build(int h[],int z[],int len){   //建树 
    if(len<=0) return -1;
    int k=find(h,z,len);
    c++;
    int f=c;
    tree[f].x=z[k];
    tree[f].l=build(h,z,k);
    tree[f].r=build(h+k,z+(k+1),len-(k+1));
    return f;
}
void bfs(int x){    //层序遍历 
    int f=0,i;
    queue<int> q;
    q.push(x);
    while(q.size()){
        int x=tree[q.front()].x;
        int l=tree[q.front()].l;
        int r=tree[q.front()].r;
        if(f==0) f=1;
        else printf(" ");
        printf("%d",x);
        if(l>-1) q.push(l);
        if(r>-1) q.push(r);
        q.pop();
    }
}
int main()
{
    int n,i,j;
    scanf("%d",&n);
    for(i=0;i<n;i++){
        scanf("%d",&h[i]);
    }
    for(i=0;i<n;i++){
        scanf("%d",&z[i]);
    }
    c=0;
    build(h,z,n);
    bfs(1);
    return 0;
}