1609. 前序和后序遍历

假设一个二叉树上所有结点的权值都互不相同。

我们可以通过后序遍历和中序遍历来确定唯一二叉树。

也可以通过前序遍历和中序遍历来确定唯一二叉树。

但是，如果只通过前序遍历和后序遍历，则有可能无法确定唯一二叉树。

现在，给定一组前序遍历和后序遍历，请你输出对应二叉树的中序遍历。

如果树不是唯一的，则输出任意一种可能树的中序遍历即可。

输入格式

第一行包含整数 N，表示结点数量。

第二行给出前序遍历序列。

第三行给出后序遍历序列。

一行中的数字都用空格隔开。

输出格式

首先第一行，如果树唯一，则输出 Yes，如果不唯一，则输出 No。

然后在第二行，输出树的中序遍历。

注意，如果树不唯一，则输出任意一种可能的情况均可。

数据范围

1≤N≤30

输入样例1：

7
1 2 3 4 6 7 5
2 6 7 4 5 3 1

输出样例1：

Yes
2 1 6 4 7 3 5

输入样例2：

4
1 2 3 4
2 4 3 1

输出样例2：

No
2 1 3 4

代码实现：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
const int N=55;
int pre[N],post[N],flag=1;
vector<int>v;
void build(int l1,int l2,int r2){
    if(l2>=r2){
        if(l2==r2)v.push_back(pre[l1]);
        return;
    }
    int temp=l2;
    while(post[temp]!=pre[l1+1])temp++;
    if(temp==r2-1)flag=0;
    build(l1+1,l2,temp);
    v.push_back(pre[l1]);
    build(l1+2+temp-l2,temp+1,r2-1);
}
int main(){
    int n;
    cin>>n;
    for(int i=0;i<n;i++)cin>>pre[i];
    for(int i=0;i<n;i++)cin>>post[i];
    build(0,0,n-1);
    if(flag)cout<<"Yes"<<endl;
    else cout<<"No"<<endl;
    for(int i=0;i<v.size();i++){
        if(i==0)cout<<v[i];
        else cout<<" "<<v[i];
    }
    cout<<endl;
    return 0;
}