1020 Tree Traversals (25 分)

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int N;
vector<int> V[30];
struct node{
    int sit=-1;//标记是否已对应过
    int num=0;//数值
    int right=-1;//右节点下标
    int left=-1;//左节点下标
};
node B[30];
int n;//记录根在中序遍历时的下标
void dfs(int v,int d)//按照每层顺序存入vector中，存入的是下标
{
    if(B[v].right==-1&&B[v].left==-1)
    return ;
    if(B[v].left!=-1)//先存入左子树的值
    {
        V[d].push_back(B[v].left);
        dfs(B[v].left,d+1);
    }
    if(B[v].right!=-1)
    {
        V[d].push_back(B[v].right);
        dfs(B[v].right,d+1);
    }
    return ;
}
int main()
{
    scanf("%d",&N);
    int A[N];
    for(int i=0;i<N;i++)
    {
        scanf("%d",&A[i]);
    }
    for(int i=0;i<N;i++)
    {
        scanf("%d",&B[i].num);
        if(B[i].num==A[N-1])//对根节点标记
        {
            B[i].sit=1;
            n=i;
        } 
    }
    int d=0;
    for(int i=N-2;i>=0;i--)//从后序数组从后往前查找
    {
        for(int j=0;j<N;j++)//从中序中找出对应值
        {
            if(B[j].num==A[i])//从该下标往两便找符合的值，记住不可跨越已经被上一次选过的值
            {
                for(int t=j+1;t<N;t++)
                {
                    if(B[t].sit==1&&B[t].left==-1)
                    {
                        B[j].sit=1;
                        B[t].left=j;
                        d=-1;
                        break;
                    }
                    else if(B[t].sit==1&&B[t].left!=-1)
                    {
                        break;
                    }
                }
                if(d!=-1)
                {
                for(int t=j-1;t>=0;t--)
                {
                    if(B[t].sit==1&&B[t].right==-1)
                    {
                        B[j].sit=1;
                        B[t].right=j;
                        break;
                    }
                    else if(B[t].sit==1&&B[t].right!=-1)
                    {
                        break;
                    }
                }
                }
                d=0;
             break;
            }
        }
    }
    printf("%d",B[n].num);
    dfs(n,1);
    for(int i=1;i<30;i++)//输出
    {
        if(V[i].size()!=0)
        {
            for(int j=0;j<V[i].size();j++)
            {
                printf(" %d",B[V[i][j]].num);
            }
        } 
    }
    return 0;
}