PAT 1119 Pre- and Post-order Traversals (30 分)

#include <cstdio>
#include <vector>
using namespace std;
vector<int> pre,in,post;
bool uni=true;//中序序列是否唯一
void getin(int prel,int prer,int postl,int postr)
{
    if(prel==prer)//前序序列只有一个元素，加入中序序列
    {
        in.push_back(pre[prel]);
        return;
    }
    if(pre[prel]==post[postr])//如果前序序列第一个元素等于后序序列最后一个元素
    {
        int i=prel+1;         //在后序序列中寻找前序序列第二个元素 
        while(i<=prer&&pre[i]!=post[postr-1])
            i++;
        if(i-prel>1)
            getin(prel+1,i-1,postl,postl+(i-prel-1)-1);
        else
            uni=false;//如果前序和后序第二个元素相同，那么根节点只有一个子树(无法确定是左是右)，故不唯一
        in.push_back(post[postr]);
        getin(i,prer,postl+(i-prel-1),postr-1);
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    pre.resize(n),post.resize(n);
    for(int i=0;i<n;i++)
        scanf("%d",&pre[i]);
    for(int i=0;i<n;i++)
        scanf("%d",&post[i]);
    getin(0,n-1,0,n-1);
    printf("%s\n%d",uni==true?"Yes":"No",in[0]);
    for(int i=1;i<in.size();i++)
        printf(" %d",in[i]);
    printf("\n");//必须有这句，否则格式错误
    return 0;
}