P1229 遍历问题

题目描述

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

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

输入格式

输A数据共两行，第一行表示该二叉树的前序遍历结果s1，第二行表示该二叉树的后序遍历结果s2。

输出格式

输出可能的中序遍历序列的总数，结果不超过长整型数。

输入输出样例

输入 #1复制

abc                           
cba

输出 #1复制

4

 

思路：

前序：根左右

后序：左右根

可知要使当前层的两种顺序指向同一棵树，有以下两种情况

前：根左   后：左根

或者

前：根右   后：右根

即：当前层只有1个儿子，存在前序AB、后序的BA，那么当前层有两种选择。

所以只要找到有几层只有1个儿子即可。

#include <bits/stdc++.h>
using namespace std;

int main()
{
    string s1,s2;
    cin>>s1>>s2;
    int ans=0;
    for(int i=0; i<s1.size(); i++){
        for(int j=1; j<s2.size(); j++){
             if(s1[i]==s2[j]&&s1[i+1]==s2[j-1]){
                ans++;
             }
        }
    }
    cout<<(1<<ans)<<endl;
    return 0;
}