HDU 6153 A Secret kmp || extkmp

题目：

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

题意：

给出两个字符串$s1,s2$，求出$s2$的每一个后缀在$s1$中出现的次数乘以这个后缀的长度，并累加求和，输出这个和

思路：

$kmp$和$extkmp$都可以做。
$extkmp$：求得是$s2$后缀在$s1$中出现的次数，我们将两个字符串翻转，就可以求$s2$前缀在$s1$中出现的次数，等价于没翻转之前求后缀，求前缀可以用$extkmp$算法，可以发现，长度较小的前缀一定包含在长度较大前缀里面，所以求出$extend$数组，记录每个公共前缀长度的出现次数，然后从长度较大的前缀开始循环，把数量一直累加到长度较小的前缀上
$kmp：$首先也是翻转两个字符串，然后匹配两个串，当$s2$串位置$j$和$s1$串位置$i$匹配上时，意味着$s2[0,j]$出现在$s1$中，也是$s2$没反转之前的一个后缀出现在$s1$中，于是统计其次数。在计算最后答案的过程中，从长度较大的后缀向较小的后缀循环，根据$Next$数组的性质，当$s2[0,i]$中的字符串出现在$s1$中时，$s2[0,Next[i]]$也一定出现$s1$中，于是把数量累加一下
$extkmp:$

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const int N = 1000000 + 10, mod = 1e9 + 7;

char ori[N], pat[N];
int Next[N], extend[N];
int num[N];
void get_next(char *pat)
{
    int len = strlen(pat);
    Next[0] = len;
    int k = 0;
    while(k + 1 < len && pat[k] == pat[k+1]) ++k;
    Next[1] = k;
    k = 1;
    for(int i = 2; pat[i]; i++)
    {
        if(i + Next[i-k] < k + Next[k]) Next[i] = Next[i-k];
        else
        {
            int j = k + Next[k] - i;
            if(j < 0) j = 0;
            while(i + j < len && pat[j] == pat[i+j]) ++j;
            Next[i] = j;
            k = i;
        }
    }
}
void extkmp(char *ori, char *pat)
{
    get_next(pat);
    int leno = strlen(ori), lenp = strlen(pat);
    int k = 0;
    while(k < leno && k < lenp && ori[k] == pat[k]) ++k;
    extend[0] = k;
    k = 0;
    for(int i = 1; ori[i]; i++)
    {
        if(i + Next[i-k] < k + extend[k]) extend[i] = Next[i-k];
        else
        {
            int j = k + extend[k] - i;
            if(j < 0) j = 0;
            while(i + j < leno && j < lenp && ori[i+j] == pat[j]) ++j;
            extend[i] = j;
            k = i;
        }
    }
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%s%s", ori, pat);
        int leno = strlen(ori), lenp = strlen(pat);
        reverse(ori, ori + leno);
        reverse(pat, pat + lenp);
        extkmp(ori, pat);
        memset(num, 0, sizeof num);
        for(int i = 0; i < leno; i++) num[extend[i]]++;
        ll ans = 0;
        for(int i = lenp; i >= 1; i--)
        {
            num[i] += num[i+1];
            ans = (ans + 1LL * num[i] * i) % mod;
        }
        printf("%lld\n", ans);
    }
    return 0;
}

$kmp:$

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
const int N = 1000000 + 10, mod = 1e9 + 7;

char ori[N], pat[N];
int Next[N];
int num[N];
void get_next(char *pat)
{
    int i = 0, j = -1;
    Next[0] = -1;
    while(pat[i])
    {
        if(j == -1 || pat[i] == pat[j]) Next[++i] = ++j;
        else j = Next[j];
    }
}
void kmp(char *ori, char *pat)
{
    get_next(pat);
    int i = 0, j = 0;
    while(ori[i])
    {
        if(j == -1 || ori[i] == pat[j]) ++i, ++j;
        else j = Next[j];
        if(j != -1) num[j]++;
        if(j != -1 && ! pat[j]) j = Next[j];
    }
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        memset(num, 0, sizeof num);
        scanf("%s%s", ori, pat);
        int leno = strlen(ori), lenp = strlen(pat);
        reverse(ori, ori + leno);
        reverse(pat, pat + lenp);
        kmp(ori, pat);
        ll ans = 0;
        for(int i = lenp; i >= 1; i--)
        {
            num[Next[i]] += num[i];
            ans = (ans + 1LL * num[i] * i) % mod;
        }
        printf("%lld\n", ans);
    }
    return 0;
}