洛谷P1229 遍历问题

遍历问题

题目描述

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

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

输入格式

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

输出格式

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

样例 #1

样例输入 #1

abc                           
cba

样例输出 #1

4

正确代码

#include <bits/stdc++.h>
using namespace std;
/**
  *author: 枫飘长安
  *date: 2024/4/15
  *fileName: P1229.cpp
  *description: TODO
  **/
typedef long long ll;
const int MAXN = 1e6+10;

int ans ;
string s1,s2;
int main (){
    cin >> s1 >> s2;
    int len1 = s1.length();
    int len2 = s2.length();

    for (int i=0; i<len1; i++) {
        for (int j=1; j<len2; j++) { // j 不能从 0开始 ,因为 0 会导致 s2[-1] 无法访问
            if (s1[i] == s2[j] && s1[i+1] == s2[j-1]) {
                ans ++;
            }
        }
    }

    cout << (1 << ans) ;
    return 0;
}