PAT甲题题解-1119. Pre- and Post-order Traversals (30)-（根据前序、后序求中序）

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string.h>

using namespace std;
const int maxn=35;
int preOrder[maxn];
int postOrder[maxn];
bool isUnique=true;
struct Node{
    int left=-1,right=-1;
}node[maxn];

/*
[preL,preR] is current sequence interval of pre-order
[postL,postR] is current sequence interval of post-order
*/
void build(int preL,int preR,int postL,int postR){
    if(preL>=preR){
        return;
    }
    int fa=preL;
    //前序遍历的第一个为根节点，第二个为子树的根节点，可能是左孩子也可能是右孩子
    int val=preOrder[preL+1];
    int postIdx;
    for(int i=postL;i<postR;i++){
        if(val==postOrder[i]){
            postIdx=i; //val在后序遍历中的索引
            break;
        }
    }
    int num=postIdx-postL; //以val为根节点的子树节点个数
    //即以val为根节点的子树只有一棵孩子，那么既可以为左孩子也可以为右孩子，所以不唯一
    if(preR-preL-1==num){
        isUnique=false;
    }
    node[fa].left=preL+1;  //不唯一的话，看做左孩子
    build(preL+1,preL+num+1,postL,postIdx);
    //如果以preL+1为根节点的子树的节点个数小于fa的所有子树节点的个数，说明fa还有右孩子
    if(preR-preL-1>num){
        node[fa].right=preL+num+2;
        build(preL+num+2,preR,postIdx+1,postR-1);
    }
}

bool first=true;
void inOrder(int root){
    if(root==-1){
        return;
    }
    inOrder(node[root].left);
    if(first){
        first=false;
        printf("%d",preOrder[root]);
    }
    else
        printf(" %d",preOrder[root]);
    inOrder(node[root].right);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&preOrder[i]);
    for(int i=1;i<=n;i++)
        scanf("%d",&postOrder[i]);
    build(1,n,1,n);
    if(isUnique)
        printf("Yes\n");
    else
        printf("No\n");
    inOrder(1);
    printf("\n");  //否则格式错误
    return 0;
}