P1229 遍历问题

遍历问题

题目描述

我们都很熟悉二叉树的前序、中序、后序遍历，在数据结构中常提出这样的问题：已知一棵二叉树的前序和中序遍历，求它的后序遍历，相应的，已知一棵二叉树的后序遍历和中序遍历序列你也能求出它的前序遍历。然而给定一棵二叉树的前序和后序遍历，你却不能确定其中序遍历序列，考虑如下图中的几棵二叉树：

所有这些二叉树都有着相同的前序遍历和后序遍历，但中序遍历却不相同。

输入格式

共两行，第一行表示该二叉树的前序遍历结果 $s_1$，第二行表示该二叉树的后序遍历结果 $s_2$。

输出格式

输出可能的中序遍历序列的总数，结果不超过 $2^{63}-1$。

样例 #1

样例输入 #1

abc                           
cba

样例输出 #1

4

#include<bits/stdc++.h`在这里插入代码片`>
using namespace std;
int flag[200];
char a[1001],b[1001];
int DFS(int sa,int ea,int sb,int eb){
	if(sa>=ea) return 1; //如果只有一个节点了，或者无节点就是一种情况 
	int len=flag[a[sa+1]]-sb+1,k=1; //len得到左子树序列节点的个数
	if(sa+len==ea) k=2;      //a[sa]这个节点只有一个子节点那么这个k==2 
	return (DFS(sa+1,sa+len,sb,sb+len-1) * DFS(sa+len+1,ea,sb+len,eb)) * k; //组合数
} 
int main(){
	scanf("%s%s",a,b);
	for(int i=0;b[i]!='\0';i++){
		flag[b[i]]=i; //标识每一个点在后序遍历序列中的什么位置 
	}
	printf("%d",DFS(0,strlen(a)-1,0,strlen(b)-1));//xu chu 
}```