（数据结构）oj重建二叉树（用权值作为新二叉树的下标）

06:由中根序列和后根序列重建二叉树

总时间限制: 

500ms 

内存限制: 

65535kB

描述

我们知道如何按照三种深度优先次序来周游一棵二叉树，来得到中根序列、前根序列和后根序列。反过来，如果给定二叉树的中根序列和后根序列，或者给定中根序列和前根序列，可以重建一二叉树。本题输入一棵二叉树的中根序列和后根序列，要求在内存中重建二叉树，最后输出这棵二叉树的前根序列。

用不同的整数来唯一标识二叉树的每一个结点，下面的二叉树

中根序列是9 5 32 67

后根序列9 32 67 5

前根序列5 9 67 32

输入

两行。第一行是二叉树的中根序列，第二行是后根序列。每个数字表示的结点之间用空格隔开。结点数字范围0～65535。暂不必考虑不合理的输入数据。

输出

一行。由输入中的中根序列和后根序列重建的二叉树的前根序列。每个数字表示的结点之间用空格隔开。

样例输入

9 5 32 67
9 32 67 5

样例输出

5 9 67 32

//该重建方法是根据中序序列和后序序列重，
//建立的二叉树的节点下标和他的权值对应,适用于权值不重复的情况 
//要注意节点在重建的序列中下标与在先序，后序序列中下标的不同 
#include<cstdio>
#include<string>
#include<sstream>
#include<algorithm>
#include<iostream>
using namespace std;

const int maxv=100000+10;
int in[maxv],post[maxv],lch[maxv],rch[maxv];	//in和post存放的是中序序列和后序序列
	//lch存放所有节点的左孩子节点标,rch存放所有节点的右孩子下标
int n;						//节点数目 

bool read_list(int *a);		//读取前序和中序表达式 

int build(int L1,int R1,int L2,int R2);
//把in[L1...R1],post[L2...R2]建成一颗二叉树，返回节点的下标 

void dfs(int u);	//二叉树的遍历 

int main()
{
	while(read_list(in))
	{
		read_list(post);
		build(0,n-1,0,n-1);			//
		dfs(post[n-1],0);
		cout<<endl; 
	}
}
 
bool read_list(int *a){		//读取前序和中序表达式 
	string line;
	if(!getline(cin,line))
		return false;
	stringstream ss(line);
	n=0;
	int x;
	while(ss>>x)			//巧妙地运用stringstream将字符串转化为数字
		a[n++]=x;
	return n>0;				
} 

int build(int L1,int R1,int L2,int R2)
{
	if(L1>R1)
		return 0;
	int root=post[R2];		//根节点的下标是后序序列最后一个数字
	int p=L1;		
	while(in[p]!=root)	p++;	//在中序序列中查找根节点在中序序列中下标，为p 
	int cnt=p-L1;					//左子树的节点个数
	lch[root]=build(L1, p-1, L2 , L2+cnt-1);	//构建左子树 
	rch[root]=build(p+1, R1, L2+cnt, R2-1);	//构建右子树 
	
	return root; 			//返回根节点下标 
}

void dfs(int u)
{
	cout<<u<<" ";
	if(lch[u])
		dfs(lch[u],sum);
	if(rch[u])
		dfs(rch[u],sum);
}