hdu 6153 A Secret【KMP／想法题】

题目

http://acm.hdu.edu.cn/showproblem.php?pid=6153

题意

给定两个串，求其中一个串 s 的每个后缀在另一个串 t 中出现的次数。

分析

后缀S(i..len)出现在后缀S(i-1…len)中，所以把两个串逆序后求KMP，这样从头扫到尾匹配一下，然后统计前缀出现的次数即可。

代码

#include <cstdio>#include <set>#include <iostream>#include <map>#include <vector>#include <cstring>#include <algorithm>#include <string>#include <cmath>using namespace  std;const int N=1e6+5;const int mod=1e9+7;typedef long long LL;char s[N],t[N];int slen,tlen;int nex[N];int num[N],vis[N];void getNext(){    int j,k;    j=0;k=nex[0]=-1;    while(j<tlen){        if(k==-1||t[j]==t[k])nex[++j]=++k;        else k=nex[k];    }}int KMP_Count(){    getNext();    int ans=0;    int j=0;    for(int i=0;i<slen;i++){        while(j&&s[i]!=t[j])j=nex[j];        if(s[i]==t[j]){            num[j]++;            j++;        }        if(j==tlen){            ans++;            j=nex[j];        }    }    return ans;}int main(){    int T;    scanf("%d",&T);    while(T--){        memset(num,0,sizeof(num));        memset(vis,0,sizeof(vis));        scanf("%s%s",s,t);        slen=strlen(s);        tlen=strlen(t);        for(int i=0;i<slen/2;i++)swap(s[i],s[slen-i-1]);        for(int i=0;i<tlen/2;i++)swap(t[i],t[tlen-i-1]);        KMP_Count();        for(int i=tlen;i>=0;i--){            if(!vis[nex[i]])num[nex[i]]+=num[i];            vis[nex[i]]=1;        }        LL ans=0;        for(int i=0;i<tlen;i++){            ans+=(LL)num[i]*(i+1);            ans%=mod;        }        cout<<ans<<endl;    }    return 0;}