【HDU3336】Count the String（kmp--每个前缀出现的次数）

题目地址：http://acm.hdu.edu.cn/showproblem.php?pid=3336

题目：

---

求给定字符串每个前缀在整个字符串出现的次数之和

 

解题思路：

---

初始答案为n，每个前缀都出现一次。

字符串的下标从0开始存，next[0]=-1，当next[i]≠0且≠-1时，说明子串的某个前缀（也是整个字符串的前缀）和整个子串的某个后缀相同，即整个字符串的前缀重复出现了一次，对答案贡献1，遍历next数组统计答案即可。

 

ac代码：

---

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 10;
const int mod = 10007;
typedef long long ll;
int nxt[maxn];
char s[maxn];
int n;
void get_nxt()
{
    int i = 0, k = -1;
    nxt[0] = -1;
    while(i < n)
    {
        if(k == -1 || s[i] == s[k])
        {
            i++;
            k++;
            nxt[i] = k;
        }
        else k = nxt[k];
    }
}
int main() {
   // freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        scanf("%s", s);
        get_nxt();
        ll ans = n % mod;
        for(int i = 0; i <= n; i++)
            if(nxt[i] != -1 && nxt[i] != 0) ans = (ans + 1) % mod;
        printf("%lld\n", ans);
    }
    return 0;
}