UVA536 二叉树重建 Tree Recovery

题意

输入一棵二叉树的先序遍历和中序遍历序列，输出它的后序遍历序列。
输入

DBACEGF ABCDEFG
BCAD CBAD

输出

ACBFGED
CDAB

思路:我们可以通过二叉树的先序和中序可以推出后序节点.先序得根,中序分左右,通过递归便可输出后序节点.
此题中只是输出后序,没有必要存储起来.

#include<bits/stdc++.h>
using namespace std;
string preorder,inorder;
void solve(string pre,string in){
	int index = in.find(pre[0]);
	if(pre==""){//递归结束条件 
		return;
	}
	solve(pre.substr(1,index),in.substr(0,index)) ;
	solve(pre.substr(index+1),in.substr(index+1));
	cout<<pre[0];
}
int main()
{
	while(cin>>preorder>>inorder){
		solve(preorder,inorder);
		cout<<endl;
	}
	
	return 0;
}

由于上面是参数传递的是字符串,中间用到了substr函数,在处理时效率较低,所以我们可以把参数改为传递索引,这样效率就大大提高了.

参考代码2

#include<bits/stdc++.h>
using namespace std;
string preorder,inorder;
void solve(int pre,int in,int len){
	if(len <= 0){//结束条件 
		return;
	}
	int index = inorder.find(preorder[pre])-in;//根节点在中序的位置索引 -- 字符串起始索引就是根节点左右子树的结点数. 
	solve(pre+1,in,index);
	solve(pre+index+1,in+index+1,len-index-1);//左右子树的节点数不一定相等,所以得用总长度-左子树结点的个数  要注意索引是从0开始的.所以最后要-1 
	cout<<preorder[pre]; 
} 
int main()
{
	while(cin>>preorder>>inorder){
		int len = preorder.size();
		solve(0,0,len);//分别指代 先序起始索引,中序起始索引,  长度 
		cout<<endl;
	}
	return 0;
}

注:substr(index,pos),从index开始截取pos个字符.如果第二个参数不写,默认截取到最后.