hdoj 6153 A Secret

题目链接：A Secret

题目大意：给你两个字符串，然后需要你对模式串的后缀去对原串匹配,原串后缀长度为1的出现多少次贡献就是1*n，长度为2的贡献就是2*n，然后要求你算出这个总贡献mod(1e9+7)

题目思路：既然要求的是后缀匹配，那么我们将两个串都翻转过来进行正着匹配就好了，然后我们需要去在KMP里面去算，当KMP匹配到失配的时候那么我们可以知道这时的j就是我们已经匹配成功的串，那么这个串的贡献是多少呢？自然是j*(j+1)/2，因为匹配了j个字符，那么j-1、j-2、…、1个字符肯定都在这之前已经匹配到了，还需要注意的一个问题是，最后失配的情况，可能一直到最后都不会失配，所以我们让其失配，然后就做一下处理就好了，这题因为j*(j+1)/2是整数，所以不需要用逆元去算，直接暴力去算就好了

#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

using namespace std;
typedef long long ll;
const int maxn = 1e6+10;
const int mod = 1e9+7;

ll n,Next[maxn],ret;
char mo[maxn],str[maxn],ans[maxn];

void getNext(){
    ll i = 0,j = -1,len = strlen(mo);
    while(i < len){
        if(j == -1||mo[i] == mo[j]) Next[++i] = ++j;
        else j = Next[j];
    }
}

void kmp(){
    ll i = 0,j = 0,len1 = strlen(mo),len2 = strlen(str);
    ll ans = 0;
    while(i < len2){
        if(j == -1||mo[j] == str[i]){
            j++;
            i++;
        }
        else ret = (ret+((j+1)*j)/2)%mod,j = Next[j];
    }
    while(j != -1){
        ret = (ret+((j+1)*j)/2)%mod;
        j = Next[j];
    }
    printf("%lld\n",ret);
}


int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        ret = 0ll;
        scanf("%s%s",str,mo);
        reverse(str,str+strlen(str));
        reverse(mo,mo+strlen(mo));
        memset(Next,-1,sizeof(Next));
        getNext();
        kmp();
    }
    return 0;
}

Uupdate:这个KMP写法有问题，有一组数据str：baabaa
mo：caabaa过不去，所以需要使用正确的姿势：

#include <bits/stdc++.h>
using namespace std;
const int N = 1000005;
const int Q = 1e9 + 7;

char s[N] , t[N];
int n , m , f[N] , cnt[N];

void work() {
    scanf("%s%s" , t , s);
    n = strlen(t);
    m = strlen(s);
    reverse(s , s + m);
    reverse(t , t + n);
    f[0] = f[1] = 0;
    for (int i = 1 ; i < m ; ++ i) {
        int j = f[i];
        while (j && s[i] != s[j])
            j = f[j];
        f[i + 1] = s[i] == s[j] ? j + 1 : 0;
    }
    memset(cnt , 0 , sizeof(int) * (m + 1));
    for (int i = 0 , j = 0 ; i < n ; ++ i) {
        while (j && t[i] != s[j])
            j = f[j];
        j += (t[i] == s[j]);
        ++ cnt[j];
    }
    for (int i = m ; i > 0 ; -- i) {
        cnt[f[i]] += cnt[i];
    }
    int res = 0;
    for (int i = 1 ; i <= m ; ++ i) {
        res += (long long)cnt[i] * i % Q;
        res %= Q;
    }
    cout << res << endl;
}

int main() {
    int T;
    scanf("%d" , &T);
    while (T --) {
        work();
    }
}