百练 / 2016计算机学科夏令营上机考试: G

题目来源：http://dsalgo.openjudge.cn/binarytree/8/

8:重建二叉树

总时间限制: 1000ms      内存限制: 65536kB

描述

给定一棵二叉树的前序遍历和中序遍历的结果，求其后序遍历。

输入

输入可能有多组，以EOF结束。
每组输入包含两个字符串，分别为树的前序遍历和中序遍历。每个字符串中只包含大写字母且互不重复。

输出

对于每组输入，用一行来输出它后序遍历结果。

样例输入

DBACEGF ABCDEFG

BCAD CBAD

样例输出

ACBFGED

CDAB

-----------------------------------------------------

解题思路

递归，首先定位根节点，再切分左子树和右子树

注意string.substr()方法的应用

-----------------------------------------------------

代码

#include <iostream>
#include <string>
using namespace std;

string postorder(string mid, string pre)			// 输入preorder字符串和inorder字符串，输出postorder字符串
{
	if(pre.length()==1) return pre;					// 如果preorder只有一个字符则preorder = inorder = postorder
	else if(pre.length()==0) return "";				// 如果preorder为空则返回空字符串
	// 这样可以避免是否存在左/右子树的讨论
	else
	{
		size_t root = mid.find(pre.at(0));			// 树的根节点
		return postorder(mid.substr(0,root), pre.substr(1,root))			// 先左子树
			+ postorder(mid.substr(root+1), pre.substr(root+1)) + pre.at(0);// 再右子树，最后根节点
		// 注意：str.substr(str.size())会返回空字符串而不会抛出异常！
	}
}

int main()
{
	string pre, mid, post;//s1为preorder traversal  s2为inorder traversal 
	while(cin >> pre >> mid)
	{
		post=postorder(mid, pre);
		cout << post << endl;
		
	}
	return 0;
}