PAT 1119. Pre- and Post-order Traversals (30) 用前序、后序求中序

// VSPAT.cpp : 定义控制台应用程序的入口点。
//


#include<cmath>
#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<stack>
#include<vector>
#include<queue>
#include<string>
#include<map>
using namespace std;

//70min
//耗时于思考和改bug
/*************************
题意：给出前序和后序，求中序
若中序不唯一，输出任意一个，并输出No
*************************/


/************************
求解要点：
对于一棵树
前序序列的首位和后序序列的末位一定相同，且为根
前序根后的那位一定为“左”子树根。
后序根前的那位一定为“右”子树。
前： 根 左子树根 XXXXXXX
后： XXXXXXX 右子树根 根

然后在后序序列中找到左子树根，左子树根的右边都是左子树
前： 根 [左子树根 XXX][右子树根XXX]
后： [XXX左子树根] [XXX右子根] 根 
根据如此划分进行递归
★若左树根=右树根，则仅1个子树，故极可能是左也可能是右
这里同意划分为左根，右孩子=-1
************************/

/***********************
笔记：
注意输入中，树的数字可能不为1~N，可能为很大的整数。
故用数组做树结构时要留意大小。
*********************/

#define M 1000 

int pre[M],post[M];
int Left[M],Right[M];
int flag = 1,n;
int buildtree(int node,int prel,int prer,int postl,int postr)
{
	int i,lchild,rchild;
	if(prel==prer||prer>=n||prel<0||postl<0||postr>=n)
	{
		Left[node]=-1;
		Right[node]=-1;
		return node;
	}
	//printf("nownode=%d\n",node);
	lchild=pre[prel+1];
	rchild=post[postr-1];
	if(lchild==rchild)
	{
		flag=0;
		Left[node]=buildtree(lchild,prel+1,prer,postl,postr-1);
		Right[node]=-1;
		//printf("node=%d,left=%d,right=%d\n",node,Left[node],Right[node]);
		return node;
	}
	for(i=postr-1;i>=postl;i--)
		if(post[i]==lchild)
			break;
	
	//左子树为 前序左到中， 后序左到中
	//右子树为 前序中到右   后序中到右
	Left[node]=buildtree(lchild,prel+1,prel+1+i-postl,postl,i);
	Right[node]=buildtree(rchild,prel+i-postl+2,prer,i+1,postr-1);

	return node;
	

	
}

int k=0;
void orderprint(int node)
{
	if(node<=0)
		return ;

	orderprint(Left[node]);

	printf("%d",node);
	k++;
	if(k<n)
		cout<<" ";
	else cout<<endl;

	orderprint(Right[node]);
}

int main()
{
	int i;
	scanf("%d",&n);
	for(i=0;i<n;i++)
		scanf("%d",&pre[i]);

	for(i=0;i<n;i++)
		scanf("%d",&post[i]);

	
	int root=pre[0];
	buildtree(root,0,n-1,0,n-1);
	if(flag==1)
		cout<<"Yes"<<endl;
	else cout<<"No"<<endl;
	orderprint(root);


	return 0;
}