PAT (Advanced Level) 1020. Tree Traversals (25)

递归建树，然后BFS一下

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;

const int maxn=;
int a[maxn],b[maxn];
int n,tot;
struct Node
{
    int left;
    int right;
    int val;
}node[maxn];

void build(int L,int R,int fa,int f)
{
    int P=-;
    for(int i=L;i<=R;i++)
        for(int j=;j<=n;j++)
            if(b[i]==a[j]) P=max(P,j);

    int root_val=a[P];

    if(tot==)
    {
        ++tot;
        node[tot].val=root_val;
    }
    else if(f==)
    {
        ++tot;
        node[tot].val=root_val;
        node[fa].left=tot;
    }
    else if(f==)
    {
        ++tot;
        node[tot].val=root_val;
        node[fa].right=tot;
    }

    int tmp=tot;

    for(int i=L;i<=R;i++)
    {
        if(b[i]==root_val)
        {
            if(i-L>) build(L,i-,tmp,);
            if(R-i>) build(i+,R,tmp,);
            break;
        }
    }

}

void bfs()
{
    queue<int>Q;
    Q.push();
    int cnt=;
    while(!Q.empty())
    {
        int head=Q.front(); Q.pop();
        printf("%d",node[head].val); cnt++;
        if(cnt<n) printf(" ");
        else printf("\n");
        if(node[head].left!=-) Q.push(node[head].left);
        if(node[head].right!=-) Q.push(node[head].right);
    }
}

int main()
{
    scanf("%d",&n); tot=;
    for(int i=;i<=n;i++) scanf("%d",&a[i]);
    for(int i=;i<=n;i++) scanf("%d",&b[i]);
    for(int i=;i<=;i++) node[i].left=node[i].right=-;
    build(,n,-,-);
    bfs();
    return ;
}