http://codevs.cn/problem/1029/
题目描述 Description
我们都很熟悉二叉树的前序、中序、后序遍历,在数据结构中常提出这样的问题:已知一棵二叉树的前序和中序遍历,求它的后序遍历,相应的,已知一棵二叉树的后序遍历和中序遍历序列你也能求出它的前序遍历。然而给定一棵二叉树的前序和后序,你却不能确定其中序遍历序列,考虑如下图中的几棵二叉树:
所有这些二叉树都有着相同的前序遍历和后序遍历,但中序遍历却不相同。
输入描述 Input Description
输入文件共2行,第一行表示该树的前序遍历结果,第二行表示该树的后序遍历结果。输入的字符集合为{a-z},长度不超过26。
输出描述 Output Description
输出文件只包含一个不超过长整型的整数,表示可能的中序遍历序列的总数。
样例输入 Sample Input
abc
cba
样例输出 Sample Output
4
数据范围及提示 Data Size & Hint
如描述
#include <iostream>
#include <string>
#include <cstring>
using namespace std;
string s1,s2,s3;
int work(int beg1,int end1,int beg2,int end2,int beg3,int end3)//beg为闭区间,end为开区间
{
int ans=0;
int len=end1-beg1;
if(len<0)return 0;
if(len==0)return 1;
if(s1[beg1]!=s3[end3-1])return 0;//由先序遍历第一个和后序遍历最后一个是否相同判断输的划分是否合理
for(int i=beg2;i<end2;i++)//以第i个数作为根节点
{
int num=i-beg2;//左边个数
int a=work(beg1+1,beg1+1+num,beg2,i,beg3,beg3+num);//递归左子树
int b=work(beg1+1+num,end1,i+1,end2,beg3+num,end3-1);//右子树
ans+=a*b;//乘法原理
}
return ans;
}
int main()
{
cin>>s1>>s3;
int len=s1.length();
cout<<work(0,len,0,len,0,len);
return 0;
}