HDU 3336 扩展kmp

 题目大意：

找到字符串中所有和前缀字符串相同的子串的个数

对于这种前缀的问题，通常通过扩展kmp来解决

其实吧这是我第一次做扩展kmp的题目，原来确实看过这个概念，今天突然做到，所以这个扩展kmp的模板是做到这道题直接copy的

这里用扩展kmp很好解决问题,_next[i]，表示第i位开始所能匹配到的最大公共前缀长度，比如说这个长度为4，那么说明前缀1,2,3,4都出现了一次，我们只在cnt[4]++

那么最后从n到1，逆向更新cnt[i] += cnt[i+1]即可，最后得到cnt[i]就表示前缀长度为i的时候所能得到的相同的子串的个数

#include <cstdio>
#include <iostream>
#include <cstring>

using namespace std;
#define N 200005
#define MOD 10007
char s[N];
int _next[N] , cnt[N];

void get_next(char *T){// _next[i]: 以第i位置开始的子串 与 T的公共前缀
     int i,length = strlen(T);
     _next[0] = length;
     for(i = 0;i<length-1 && T[i]==T[i+1]; i++);
          _next[1] = i;
          int a = 1;
          for(int k = 2; k < length; k++){
                  int p = a+_next[a]-1, L = _next[k-a];
                  if( (k-1)+L >= p ){
                       int j = (p-k+1)>0? (p-k+1) : 0;
                       while(k+j<length && T[k+j]==T[j]) j++;// 枚举(p+1，length) 与(p-k+1,length) 区间比较
                       _next[k] = j, a = k;
                  }
                  else _next[k] = L;
         }
}
int main()
{
   // freopen("a.in" , "r" , stdin);
    int T;
    scanf("%d" , &T);
    while(T--)
    {
        int n;
        scanf("%d%s" , &n , s);
        get_next(s);
      //  for(int i=0 ; i<n ; i++) cout<<_next[i]<<endl;
        memset(cnt , 0 , sizeof(cnt));
        for(int i=0 ; i<n ; i++) cnt[_next[i]]++;
        for(int i=n-1 ; i>=1 ; i--) cnt[i]+=cnt[i+1];
        int sum=0;
        for(int i=1 ; i<=n ; i++) sum = (sum+cnt[i])%MOD;
        printf("%d
" , sum);
    }
    return 0;
}