HDU 3336 Count the string（KMP）

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

解题思路：

先把原数组构造失配数组

fail[i] 表示0~i-1最长相同前后缀

首先我们先令ans = len，也就是先把每个前缀本身加上去

我们可以保证在fail数组所得到的后缀不会与这些重合，不然那叫前缀。

接下去遍历fail[2]~fail[len]

对于每个fail我们求出所有以i-1为后缀结尾的后缀

首先当前fail[i]不为0的话，一定存在0 ~ x  == i-1-x ~ i-1

并且我们保证这个后缀之前没有被计数，因为之前没有得到过i-1为结尾的后缀

我们要得到所有以i-1为后缀的要怎么做，就是沿着失配边回去，看fail[fail[i]]是否大于0，大于0证明还可以从这个i-1-x~i-1再抠出一个以i-1结尾的更短的后缀，再看fail[fail[fail[i]]]直到这个值不再大于0.

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define mod 10007

using namespace std;

const int N = 2e5+5;

char s[N];
int fail[N];

void kmp(int len)
{
    fail[0] = -1;
    for (int i=0,j=-1;i<len;){
        if (j==-1||s[i]==s[j]){
            i++;j++;
            fail[i] = j;
        }
        else j = fail[j];
    }

    int ans = len%mod;
    for (int i=2;i<=len;i++){
        if (fail[i]>0) ans++;
        int now = fail[i];
        while (fail[now]>0){
            ans++;
            now = fail[now];
        }
    }
    printf("%d\n",ans%mod);
}

int main()
{
    int T;
    scanf("%d",&T);
    while (T--){
        int n;
        scanf("%d",&n);
        scanf("%s",s);
        kmp(n);
    }
    return 0;
}

也不知道我这么想是不是复杂了，以后如果从其他题感受到了更简单的方法就回来改一下