1364：二叉树遍历(flist)

题目

1364：二叉树遍历(flist)

时间限制: 1000 ms 内存限制: 65536 KB
【题目描述】
树和二叉树基本上都有先序、中序、后序、按层遍历等遍历顺序，给定中序和其它一种遍历的序列就可以确定一棵二叉树的结构。

假定一棵二叉树一个结点用一个字符描述，现在给出中序和按层遍历的字符串，求该树的先序遍历字符串。

【输入】
两行，每行是由字母组成的字符串（一行的每个字符都是唯一的），分别表示二叉树的中序遍历和按层遍历的序列。

【输出】
一行，表示二叉树的先序序列。

【输入样例】
DBEAC
ABCDE
【输出样例】
ABDEC

题目分析：层次遍历第一个字符即树的根，我们知道树的根后，从中序遍历中查找这个字符，它左面的即左子树的中序遍历，右面的即右子树的中序遍历，然后从层次遍历中分别删去与左右子树中序遍历不同的字符，就能确定出左右子树的根，然后对左右子树分别递归即可。

C++代码

#include<iostream>
using namespace std;
void func(string a, string b)
{
	if (!a.size())//为空
		return;
	cout << b[0];//层次遍历第一个字符就是根
	int pos = a.find(b[0]);
	string a1 = a.substr(0, pos), b1;//a1为左子树的中序遍历，b1用于确定左子树的根
	for (int i = 0; i < b.size(); i++)
	{
		if (a1.find(b[i]) != string::npos)
			b1.push_back(b[i]);
	}
	func(a1, b1);//求左子树
	a1 = a.substr(pos + 1);//同上
	b1.clear();
	for (int i = 0; i < b.size(); i++)
	{
		if (a1.find(b[i]) != string::npos)
			b1.push_back(b[i]);
	}
	func(a1, b1);//求右子树
}
int main()
{
	ios::sync_with_stdio(false);
	string a, b;
	cin >> a >> b;
	func(a, b);
	return 0;
}