HDU6153（KMP）

题解:两个字符串转置后倒着跑一遍KMP，最后求这个前缀在匹配串中有几个相加一下即可，接着就是ans[i]*i的总和相加

#include <map>#include <queue>#include <cmath>#include <cstdio>#include <complex>#include <cstring>#include <cstdlib>#include <iostream>#include <algorithm>using namespace std;typedef long long int ll;const int N=1e6+10;const int MOD=1e9+7;char a[N],b[N];int f[N];ll ans[N];void get_next(){    int m=strlen(b);    f[0]=f[1]=0;    for(int i=1;i<m;i++){        int j=f[i];        while(j&&b[i]!=b[j]) j=f[j];        f[i+1]=b[i]==b[j]?j+1:0;    }}void kmp_count(){    int n=strlen(a),m=strlen(b);    int j=0;    for(int i=0;i<n;i++){        while(j&&b[j]!=a[i]) j=f[j];        if(b[j]==a[i]) j++;        ans[j]++;        if(j==m){            j=f[j];            if(j>m) j=0;        }    }}int main(){int n;    scanf("%d",&n);    while(n--) {        scanf("%s%s",a,b);        reverse(b,b+strlen(b));        reverse(a,a+strlen(a));        memset(ans,0,sizeof(ans));        get_next();kmp_count();        int m=strlen(b);        for(int i=m;i>=1;i--) ans[f[i]]+=ans[i];        ll sum=0;        for(ll i=1;i<=m;i++){           sum=(sum+i*ans[i])%MOD;        }        printf("%lld\n",sum);    }    return 0;}