hdu 6153 A Secret(KMP)

题目链接：hdu 6153 A Secret

题意：

给你两个字符串a,b，问你对于b的每个后缀在a中出现了多少次，然后输出sum{每个后缀的长度*该后缀在a中出现的次数}。

题解：

将a,b反转一下，然后跑一下kmp，在途中记录一下哪些位置匹配到了。

然后再倒着统计一下答案就行了。

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;
typedef long long ll;
const int N=1e6+7,P=1e9+7;
int t,lena,lenb,nxt[N],vis[N],ans[N];
char a[N],b[N];

void KMP(int n, char*a, int m, char*b) {
    int i, j;
    for (nxt[0] = j = -1, i = 1; i < n; nxt[i++] = j) {
        while (~j&&a[j + 1] != a[i])j = nxt[j];
        if (a[j + 1] == a[i])j++;
    }
    for (j = -1, i = 0; i < m; i++) {
        while (~j&&a[j + 1] != b[i])j = nxt[j];
        if (a[j + 1] == b[i])j++;
        if(j!=-1)vis[j]++;
        if (j == n - 1)j = nxt[j];
    }
}

int main(){
    scanf("%d",&t);
    while(t--)
    {
        mst(vis,0),mst(ans,0),scanf("%s%s",a,b);
        lena=strlen(a),lenb=strlen(b);
        reverse(a,a+lena);
        reverse(b,b+lenb);
        KMP(lenb,b,lena,a);
        for(int i=lenb;i;i--)
        {
            ans[i]+=vis[i-1];
            if(nxt[i-1]!=-1)ans[nxt[i-1]+1]+=ans[i];
        }
        int ret=0;
        F(i,1,lenb)ret=(ret+(ll)i*ans[i])%P;
        printf("%d\n",ret);
    }
    return 0;

}