HDU 3336 数据结构之KMP

点击打开链接

题意：t组数据，每组开头一个n，然后一个长度为n的字符串，问以第一个为串的数量到以整个串为串的数量的和，最后对10007取余
思路：用KMP的next数组保存就行了next数组存的是与开头进行匹配的可以匹配到的位置，如abab，则next数组为0 0 1 2，第三个的1表示从头匹配的匹配到的位置，第四个数就是4了，然后从头到尾扫一遍，直到next数组为0为止，每一次加1

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=200010;
const int mod=10007;
char str1[maxn];
int Next[maxn];
void makenext(int m){
    int i=0,j=-1;
    Next[i]=-1;
    while(i<m){
        if(j==-1||str1[i]==str1[j])
            Next[++i]=++j;
        else j=Next[j];
    }
}
//int KMP(int n,int m){
//    int i=0,j=0,ans=0;
//    while(i<n){
//        if(j==-1||str2[i]==str1[j]) i++,j++;
//        else j=Next[j];
//        if(j==m){
//            ans++;j=Next[j-1];i--;
//        }
//    }
//    return ans;
//}
int main(){
    int T,n;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        scanf("%s",str1);
        makenext(n);
        int sum=0;
        for(int i=1;i<=n;i++){
            int t=i;
            while(t){
                sum=(sum+1)%mod;
                t=Next[t];
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}