A1119 Pre- and Post-order Traversals (30) 前序后序转中序

原创于 2020-07-15 19:15:02 发布 · 167 阅读

0 ·

CC 4.0 BY-SA版权

pat刷题专栏收录该内容

32 篇文章

订阅专栏

本文探讨了基于先序和后序遍历序列重构二叉树的算法，通过递归方式确定中序遍历序列，进而恢复二叉树结构。代码实现采用C++，并讨论了边界条件处理。

本题前导：https://blog.youkuaiyun.com/destiny_balabala/article/details/99352715；
本题假设是如果判断不了，设该结点为右边的结点。

#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
const int maxn=30;
bool flag=false;
int post[maxn],pre[maxn];
vector<int> ans;
void getin(int prel,int prer,int postl,int postr)
{
	if(prel==prer)
	{
		ans.push_back(pre[prel]);
		return ;
	}
		int i=prel+1;
		while(pre[i]!=post[postr-1])i++;
		if(i-prel-1>0)
		 getin(prel+1,i-1,postl,postl+i-2-prel);
		else
		  flag=true;
		  ans.push_back(post[postr]);
		 getin(i,prer,postl+i-1-prel,postr-1);
}
int main()
{
	int n;
	cin>>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);
	if(flag==true)
	printf("No\n");
	else
	printf("Yes\n");
	for(int i=0;i<ans.size();i++)
	{
		printf("%d",ans[i]);
		if(i!=n-1)
		printf(" ");
		else 
        printf("\n");
	}
}