pta 甲级1119 Pre- and Post-order Traversals (30分)

链接：https://pintia.cn/problem-sets/994805342720868352/problems/994805353470869504

题意：给出二叉树的前序、后序遍历序列，能否唯一确定一棵树？输出中序遍历序列。

思路：能否确定一棵树关键在于是否有只有一个孩子的节点。如果有，那么就无法确定唯一的树。至于怎么确定中序遍历序列，可以先根据后序序列确定右孩子，通过前序序列确定左右子树。递归解决，详情看代码注释。

#include <bits/stdc++.h>
#define ll long long
#define pb push_back
using namespace std;
const int N = 100;
int pr[N],po[N],n;
vector<int> in;
bool flag=1;
//通过前序来确定节点关系 
void solve(int prel,int prer,int postl,int postr){
	//如果是叶子节点，直接返回 
	if(prel==prer){
		in.pb(pr[prel]);
		return ;
	}
	//说明新确定一个父节点 
	if(pr[prel]==po[postr]){
		//假定右孩子是postr-1，找左孩子 
		int pos=prel+1;
		while(pos<=prer&&pr[pos]!=po[postr-1]) pos++;
		//如果只有一个孩子，无法确定是左孩子还是右孩子，不能唯一确定一棵树 
		if(pos-prel==1)
			flag=0;
		else
		//确定左子树顺序，前序范围自然变为prel+1~pos-1，后序范围则变为postl~postl+pos-(prel+1)-1，因为左子树的大小一定  
		//所以前序、后序区间大小相等
			solve(prel+1,pos-1,postl,postl+pos-1-(prel+1));	
		//中序遍历，左、父、右 
		in.pb(po[postr]);
		//确定右子树顺序，前序范围自然为pos~prer，后序范围则变为postl+pos-(prel+1)~postr-1，因为右子树的大小一定
		//所以前序、后序区间大小相等 		
		solve(pos,prer,postl+pos-(prel+1),postr-1);
	}
}
int main(void){
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>pr[i];
	for(int i=1;i<=n;i++)
		cin>>po[i];
	solve(1,n,1,n);
    puts(flag?"Yes":"No");
    int siz=in.size();
    for(int i=0;i<siz;i++){
    	cout<<in[i];
    	if(i!=siz-1)
    		cout<<" ";
	}
	cout<<endl;
	return 0;
}