树的遍历---pat

树的遍历 (25分)

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式：
输入第一行给出一个正整数
NN
N（
≤30\le 30
≤30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式：
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。
输入样例：
7
2 3 1 5 7 6 4
1 2 3 4 5 6 7
输出样例：
4 1 6 3 5 7 2

已知后序与中序输出前序（先序）：
后序：3, 4, 2, 6, 5, 1（左右根）
中序：3, 2, 4, 1, 6, 5（左根右）
分析：因为后序的最后一个总是根结点，令i在中序中找到该根结点，则i把中序分为两部分，左边是左子树，右边是右子树。因为是输出先序（根左右），所以先打印出当前根结点，然后打印左子树，再打印右子树。左子树在后序中的根结点为root – (end – i + 1)，即为当前根结点-右子树的个数。左子树在中序中的起始点start为start，末尾end点为i – 1.右子树的根结点为当前根结点的前一个结点root – 1，右子树的起始点start为i+1，末尾end点为end。
输出的前序应该为：1, 2, 3, 4, 5, 6（根左右）

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

struct node
{
    int va , pos ;
}la[32];

int main()
{
    queue<node> s;
    int mi[32] , re[32];
    int n ;
    bool vis[32];
    cin >> n ;
    if(n == 0)
        return 0 ;
    for(int i = 1 ; i <= n ; ++ i)
        cin >> la[i].va,la[i].pos=i;
    for(int i = 1 ; i <= n ; ++ i)
        cin >> mi[i],re[mi[i]]=i;
    memset(vis,0,sizeof(vis));
    s.push(la[n]);
    int ans = 1 ;
    while(ans != n)
    {
        node ro = s.front();
        s.pop();
        int cnt = 0 ;
        vis[re[ro.va]] = true ;
        for(int i = re[ro.va]+1 ; i <= n ; i ++)
        {
            if(vis[i] || i > n)
                break ;
            cnt ++ ;
        }

        int cnt2 = 0 ;
         for(int i = re[ro.va]-1 ; i >= 0 ; i --)
        {
            if(vis[i] || i == 0)
                break ;
            cnt2 ++ ;
        }

        if(ro.pos-cnt-1 >= 1 && cnt2 != 0)
            s.push(la[ro.pos-cnt-1]); // 压入左树root

        if(ro.pos-1 >= 1 && cnt != 0)
            s.push(la[ro.pos-1]); // 压入右树root
        ans ++ ;
        printf("%d ",ro.va);
    }
    node last = s.front();
    printf("%d\n",last.va);
    //printf("\n");
    return 0 ;
}