BZOJ[3620]似乎在梦中见过的样子 KMP

传送门ber~

这题…不是和BZOJ[3670]那题一样吗？

对每一个位置进行两遍KMP，第一遍求出nex数组后再进行一遍以统计合法答案
比那题多个判断条件…其余一模一样

代码如下：

#include<cstring>
#include<cstdio>
#define MOD 1000000007
#define N 1000020
using namespace std;
char s[N];
int nex[N],num[N],T,k,ans;
inline void Solve(char *s){
    int len=strlen(s+1);
    nex[1]=0;num[1]=1;
    int i=2,j=0;
    while(i<=len){
        if(!j || s[i]==s[j+1]){
            if(s[i]==s[j+1]) j++;
            nex[i]=j;
            num[i]=num[j]+1;
            i++;
        }
        else j=nex[j];
    }
    i=2,j=0;
    while(i<=len){
        if(!j || s[i]==s[j+1]){
            if(s[i]==s[j+1]) j++;
            while(j*2>=i) j=nex[j];
            if(j>=k) ans++;
            i++;
        }
        else j=nex[j];
    }
}
int main(){
    scanf("%s",s+1);
    scanf("%d",&k);
    int len=strlen(s+1);
    for(int i=1;i<=len;i++)
        Solve(s+i-1);
    printf("%d",ans);
return 0;
}