hdu 6153 A Secret (KMP)

题目链接

因为要求字符串的所有后缀和另一个串的匹配，和kmp中匹配时求next数组有相似之处。可以把先把他们做一个转置的处理，再跑一遍kmp。同时用一个num数组来记录前缀长度为i时的匹配次数。但是我们这时求得的num并不是最终答案，要得到正确的num还需要一个累加的过程。即num[next[i]]+=num[i]。
然后就可以求得答案了。

#include <iostream>
#include <cstring>
#include <cstdio>
#define ll long long
using namespace std;

const int maxn= 1e6+10;
const int mod=1e9+7;

int nxt[maxn];
char S[maxn], T[maxn];
int slen, tlen;
ll num[maxn];
void getNext()
{
    int j, k;
    j = 0;
    k = -1;
    nxt[0] = -1;
    while(j < tlen)
        if(k == -1 || T[j] == T[k])
            nxt[++j] = ++k;
        else
            k = nxt[k];
}

/*
返回模式串在主串S中出现的次数
*/
void KMP_Count()
{
    int i, j = 0;
    memset(num,0,sizeof(num));

    getNext();
    for(i = 0; i < slen; i++)
    {
        while(j > 0 && S[i] != T[j])
            j = nxt[j];
        if(S[i] == T[j])
        {
            j++;
            num[j]++;
        }
        if(j == tlen)
        {
            j = nxt[j];
        }
    }
}
int t;
char a[maxn],b[maxn];
int main()
{
    cin>>t;
    while(t--)
    {
        scanf("%s",a);
        slen=strlen(a);
        for(int i=0; i<slen; i++)
            S[i]=a[slen-1-i];
        scanf("%s",b);
        tlen=strlen(b);
        for(int i=0; i<tlen; i++)
            T[i]=b[tlen-1-i];
        KMP_Count();
        ll ans=0;
        for(int i=tlen;i>=1;i--)
        {
            num[nxt[i]]+=num[i];
         //   cout<<num[i]<<endl;
            ans=(ans+i*num[i])%mod;
        }
        cout<<ans<<endl;


    }


    return 0;
}