HDU 6153 A Secret（kmp）

最新推荐文章于 2020-02-07 11:05:28 发布

Konjac.k

最新推荐文章于 2020-02-07 11:05:28 发布

阅读量261

点赞数

分类专栏： ACM —— 思维

本文链接：https://blog.csdn.net/addkai/article/details/77450051

版权

ACM —— 思维专栏收录该内容

13 篇文章 0 订阅

订阅专栏

题目：点击打开链接

题意：给两个字符串S1，S2，求S2中每个后缀子串在S1中出现的次数*该后缀子串长度，最后将所有这些求和。

思路：kmp是复杂度O(m+n)的较快的字符串匹配算法，但如果取出每个后缀子串进行字符串匹配的话是O(m(n+m))也就是O(n^2)的复杂度，则超时。所以该题只能一遍kmp找到并记录S2中每个后缀子串出现的次数，这就要求对kmp的内部运行十分了解。首先kmp是从头往后进行比对，所以题目给的S1，S2需要全部倒置后再kmp。对于kmp的相关改动为：枚举时的主串长度+1，j调到next[j]之前都要用数组a记录长度为j的匹配长度a[j]++，最后从后往前处理数组a，a[i]+=a[i+1]，就得到了S2中每个后缀子串在S1中出现的次数a[i]，再乘上每个后缀子串的长度，也就是i，相加取余得ans。

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int N = 1e6+6;
int nex[N];
char S[N], T[N];
int slen, tlen;
int a[N];
void getNext()
{
    int j, k;
    j = 0;
    k = -1;
    nex[0] = -1;
    while(j < tlen)
        if(k == -1 || T[j] == T[k])
            nex[++j] = ++k;
        else
            k = nex[k];

}
void KMP_Count()
{
    int i, j = 0;
    if(slen == 1 && tlen == 1)
    {
        if(S[0] == T[0])
        {
            a[1]=1;
            return;
        }
        else
            return;
    }
    getNext();
    for(i = 0; i <= slen; i++)
    {
        while(j > 0 && S[i] != T[j])
        {
            a[j]++;
            j = nex[j];
        }
        if(S[i] == T[j])
        {
            j++;
        }
        if(j == tlen)
        {
            a[j]++;
            j = nex[j];
        }
    }
    return;
}
int main()
{
    int _T;
    scanf("%d",&_T);
    while(_T--)
    {
        scanf("%s%s",S,T);
        slen=strlen(S);
        tlen=strlen(T);
        reverse(S,S+slen);
        reverse(T,T+tlen);
        memset(a,0,sizeof(a));
        KMP_Count();
        ll ans=0;
        for(ll i=tlen; i>=1; i--)
        {
            a[i]=(a[i]+a[i+1])%mod;
            ans=(ans+a[i]*i%mod)%mod;
        }
        printf("%lld\n",ans);
    }
    return 0;
}