[kuangbin带你飞] - Count the string （KMP）

题目链接： K - Count the string HDU - 3336

题意

​ 给你一个字符串，枚举其所有的前缀，输出其所有前缀出现的次数

---

思路

1. 题目要求的是给定一个字符串s，求字符串s的所有的前缀在s的匹配的次数之和mod10007.
2. 很明显n<= 200000，分析一下那么就要n个前缀如果每一个前最都去匹配s的话复杂度就是o(n^2)，那么肯定是TLE的，所以要考虑另外的思路
3. 这个与Next数组一定有千丝万缕的关系，Next[i]储存的是在第i位的最大前后缀数，那么我现在需要的就是从n->1，例如ababa，在Next[4]的时候，这里重复出现的就不只是当前aba了，而是aba和a,所有我需要不断向前j = Next[j-1]，然后每访问一次，就使ans++
4. 以上的算法还不够完美，这里可以设置一个DP数组来储存当前的位数的值，那么每次只需向前返回一次就足以。

---

代码

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)(j);i <= (int)(k);i ++)
#define per(i,j,k) for(int i = (int)(j);i >= (int)(k);i --)
#define mmm(a,b)   memset(a,b,sizeof(a))
#define push_back pb

typedef long long ll;
const int INF = (int)0x3f3f3f3f;
const int MAXN = (int)2e5+7;
const int MOD = (int)1e4+7;

char sub[MAXN];
int Next[MAXN];
int dp[MAXN];
int len1;

void MakeNext(){
    Next[0] = 0;
    for (int i = 1,j = 0;i < len1;i ++){
        while (j > 0 && sub[i] != sub[j]) j = Next[j-1];
        if (sub[j] == sub[i]) j ++;
        Next[i] = j;
    }
}

int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

    int T;
    cin >> T;
    while (T --){
        cin >> len1 >> sub;
        MakeNext();
        dp[0] = 0;
        int ans = 0;
        rep(i,1,len1){
            dp[i] = dp[Next[i-1]] + 1;
            ans = (ans + dp[i])%MOD;
        }
        cout << ans << endl;
    }
}