【洛谷】P1229 遍历问题

洛谷P1229 遍历问题

题目描述

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

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

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

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

输入输出样例
输入 #1

abc
cba

输出#1

4

之前做前序后序遍历求中序的时候其实还以为这是唯一的。。这题上来我是不会做的，没想到什么好的方法。但实际上分析样例可以发现，当一个节点只有一个儿子的时候，不管是左孩子还是右孩子，前序遍历是AB，后序遍历就是BA。在样例中，有两个节点是这样，所以就是2*2=4种情况。

所以等价于寻找一个孩子的节点。一个孩子的节点的特征是，前序遍历里是AB，后序遍历里也是连续的BA。所以代码就出来了，想到了就很简单：

#include <iostream>
#include <limits.h>
#include <cstdio>
#include <cmath>
#include <stack>
#include <string>
#include <algorithm>
#include <sstream>
#include <vector>
#include <queue>
#include <cstring>
#include <map>
#include <list>
#include <set>
using namespace std;
long res=1;
string pre,post;
int main(){
    cin>>pre>>post;
    int n=pre.size();
    for(int i=0;i<n;i++){
        for(int j=1;j<n;j++){
            if(pre[i]==post[j]&&pre[i+1]==post[j-1]){
                res*=2;
            }
        }
    }
    cout<<res;
    return 0;
}