hdu-6153 A Secret

最新推荐文章于 2020-08-26 14:16:37 发布

ACpartner

最新推荐文章于 2020-08-26 14:16:37 发布

阅读量406

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/ACpartner/article/details/77461166

本文介绍了一种使用KMP算法解决特定字符串匹配问题的方法，即求解第二个字符串的所有后缀在第一个字符串中出现次数的加权总和。通过反转字符串并利用KMP算法的特性，文章详细阐述了如何高效地进行匹配计数。

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

题意：给两个字符串，求第二个字符串的所有后缀在第一个字符出现次数的次数*长度的总和。

思路：将两个串反转，则变成了求第二个串的所有前缀在第一个字符出现的次数。此时：我们可以用KMP匹配两个串，失配时的位置的之前位置必然是已经匹配好的。那么：我们只要在失配的位置 i 或者整个串匹配完的位置给累加答案，代表 0 - i位置都匹配好一次。然后最后再去处理答案。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <math.h>
#define LL long long
#define siz 1000005

using namespace std;
const int mod = 1e9+7;
char str[2*siz],s[siz];
int nex[siz];
int ans[siz];
void kmp_pre(char x[],int m){
    int i,j;
    j = nex[0] = -1;
    i = 0;
    while(i < m){
        if(j==-1 || x[i] == x[j])
            nex[++i] = ++j;
        else j = nex[j];
    }
}
void KMP_Count(char x[],int m,char y[],int n){
    int i,j;

    kmp_pre(x,m);
    i = j = 0;
   /* for(int i=0;i<m;i++){
        cout<<nex[i]<<endl;
    }*/
    for(i = 0; i < n; i++)
    {
        while(j && y[i] != x[j])
        {
           // cout<<i<<"---"<<j<<endl;
            ans[j]++;
            j = nex[j];
        }
        if(y[i] == x[j])
        {
            j++;
        }
        if(j == m)
        {
            //cout<<i<<"&&&"<<j<<endl;
            ans[j]++;
            j = nex[j];
        }
    }
   // cout<<i<<"^^^"<<j<<endl;
    //if(j<m) ans[j] += 1;
}
void solve(){
    int len1 = strlen(str);
    int len2 = strlen(s);
    /*for(int i=0;i<len1/2;i++){
        swap(str[i],str[len1-i-1]);
    }*/
    strrev(str);
    strrev(s);
    for(int i=0;i<len2;i++){
        str[i+len1] = '-';
    }
    str[len2+len1] = '\0';
    len1 = len2 + len1;
    /*for(int i=0;i<len2/2;i++){
        swap(s[i],s[len2-i-1]);
    }*/
    ///puts(str);
    //puts(s);
    KMP_Count(s,len2,str,len1);
    LL res = 0;
    LL r = 0;
    for(int i=len2;i>=0;i--){
        //cout<<i<<" "<<ans[i]<<endl;
        r =(r + ans[i])%mod;
        res = (res + (((1ll*i)%mod)*(r%mod)%mod)) % mod;
    }
    printf("%lld\n",res);
}
int main()
{
    int T;
    scanf("%d\n",&T);
    while(T--){
        memset(ans,0,sizeof(ans));
        scanf("%s",str);
        scanf("%s",s);
        solve();
    }
    return 0;
}