[测试题]無名（noname）

Description

因为是蒯的题所以没想好名字，为什么要用繁体呢？去看《唐诗三百首》吧！

题意很简单，给你一个串，求他有多少个不同的子串，满足前缀为A，后缀为B。

需要注意的是，串中所有的字母都是小写字母。

Input

一共3行。

第一行母串S；

第二行串A；

第三行串B。 

Output

一个数，即有多少不同的子串。

Sample Input

abababab

a

b

Sample Output

4

Hint

100%:

length(S)<=2000;

length(A)<=2000;

length(B)<=2000;

30%:都少个0。

题解

我们可以预处理出所有$A$串和$B$串在$S$中的位置。

找到合法的串，我们可以去找前缀和后缀的位置，取出中间的部分用$hash$判重就好了。

 

#include<set>
#include<map>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<vector>
#include<cstdio>
#include<string>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
const LL N=2000;
const LL MOD=1e8+7;

LL Max(const LL &a,const LL &b) {return a>b ? a:b;}
LL Min(const LL &a,const LL &b) {return a<b ? a:b;}

char S[N+5],A[N+5],B[N+5];
LL ls,la,lb;
bool vis[N+5];
LL ans;
LL q[N+5],hash[N+5];
bool pd[MOD+5];

bool find_hash(LL x)
{
    if (pd[x]) return false;
    pd[x]=1;
    return true;
}

int main()
{
    S[0]=A[0]=B[0]='$';
    scanf("%s%s%s",S+1,A+1,B+1);
    ls=strlen(S)-1,la=strlen(A)-1,lb=strlen(B)-1;
    q[0]=1;
    for (LL i=1;i<=ls;i++) q[i]=(q[i-1]*31)%MOD;
    for (LL i=1;i<=ls;i++) hash[i]=(hash[i-1]*31+S[i]-'a'+1)%MOD;
    for (LL i=1;i<=ls;i++) if (S[i]==B[1])
    {
        LL j;
        for (j=2;j<=lb;j++) if (S[i+j-1]!=B[j]) break;
        if (j>lb) vis[i]=1;    
    }
    for (LL i=1;i<=ls;i++) if (S[i]==A[1])
    {
        LL j;
        for (j=2;j<=la;j++) if (S[i+j-1]!=A[j]) break;
        if (j>la) 
        {
            for (LL j=Max(i,la-lb+i);j<=ls;j++) if (vis[j])
            {
                LL r=j+lb-1;
                if (find_hash((hash[r]+MOD-hash[i-1]*q[r-i+1]%MOD)%MOD)) ans++;
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}