HDU 6153 A Secret（扩展kmp模板题）_hdu kmp 模板题-优快云博客

本文链接：https://blog.youkuaiyun.com/ThirteenTime/article/details/77448553

本文介绍了一种基于扩展KMP（extendKMP）的字符串匹配算法，该算法用于计算一个字符串的所有后缀在另一个字符串中出现的次数与后缀长度的乘积之和。通过反转字符串并利用ex数组记录最大匹配前缀长度，最终通过求和函数得出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给你两个字符串，求B的每一个后缀在A中出现的次数与其长度之积。

可以使用extend kmp求得ex【i】（表示对于B与subA（i，strlen（A）-1）的最大匹配前缀）；所以我们先反置两个字符串，之后，对于每一个得到的ex【i】，他可以带来的变化是（1+2+3+4+……+i）；所以加一个求和函数来得到答案，注意我代码中的ex【i】为int型，在Add函数中，我们如果还是传入int型的话，在运算中会炸。。。转化成longlong就行了

#include<bits/stdc++.h>
using namespace std;

#define maxn 1000001
#define mod 1000000007
typedef long long LL;
LL ans;

int t;
char str1[maxn],str2[maxn];
int Next[maxn],ex[maxn];

void getNext(char str[])
{
    int i = 0, j, po, len = strlen(str);
    Next[0] = len;
    while(str[i] == str[i+1]
        &&i + 1 < len) i++;
    Next[1] = i; po = 1;
    for(i = 2; i < len; i++)
    {
        // 
        if(Next[i - po] + i < Next[po] + po)
            Next[i] = Next[i - po];
        else 
        {
            j = Next[po] + po - i;
            if(j < 0) j = 0;
            while(i + j < len&&str[j] == str[j + i])
                j++;
            Next[i] = j;
            po = i;
        }
    }
}

void ex_kmp(char str1[], char str2[])
{
    int i = 0, j, po, len1 = strlen(str1);
    int len2 = strlen(str2);
    getNext(str2);
    while(str1[i] == str2[i]&&
        i < len2&&i < len1) i++;
    ex[0] = i;
    po = 0;
    for(int i = 1; i < len1; i++)
    {
        if(Next[i - po] + i < ex[po] + po)
            ex[i] = Next[i - po];
        else 
        {
            j = ex[po] + po - i;
            if(j < 0) j = 0;
            while(i + j < len1&&j < len2
            &&str1[i + j] == str2[j]) j++;
            ex[i] = j;
            po = i;
        }
    }
}

LL add(LL x)
{
    return (x*(x+1))/2%mod;
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        ans = 0;
        scanf("%s%s", str1, str2);
        int len1 = strlen(str1);
        int len2 = strlen(str2);
        reverse(str1,str1+len1);
        reverse(str2,str2+len2);
        ex_kmp(str1, str2);
        for(int i = 0; i < len1; i++)
        {
            ans = ans + add(ex[i]);
            ans %= mod;
        }
        printf("%lld\n", ans);
    }
}