二叉树的每个结点用一个大写字母表示,现在已知其中序和按层遍历的字符串,求该树的先序遍历字符串。
输入格式:
输入两行,分别表示二叉树的中序遍历和按层遍历的序列(一行中的每个字母都不相同)。
输出格式:
一行,表示二叉树的先序序列。
输入样例:
DBEAC
ABCDE
输出样例:
ABDEC
代码
#include<iostream>
#include<cstring>
#include<string>
using namespace std;
//因为层序是先输出根节点的,所以可以递归查找
string s1,s2;
void find(int l1,int r1,int l2,int r2)
{
int i,j;
for(i=l2;i<=r2;i++)//找层次遍历中优先输出根节点的位置
{
int f=0;
for(j=l1;j<=r1;j++)
{
if(s2[i]==s1[j])//输出根节点
{
cout<<s1[j];
f=1;
break;
}
}
if(f)
break;
}
if(j>l1)
find(l1,j-1,0,r2);//遍历左子树
if(j<r1)
find(j+1,r1,0,r2);//遍历右子树
}
int main()
{
cin>>s1>>s2;
find(0,s1.length()-1,0,s2.length());
return 0;
}