题意:
给出一个字符串,要求求出形如A+B+A的子串数量,且lenA≥k,lenB≥1。字符串长度≤15000,k≤100,所以字符长度为小写字母。
题解:
第一次写kmp的题QAQ~这题利用的是fail函数的性质:若字符串s在位置x的fail函数f[x]不为0,则prefix(s+1,s+x)的长度为f[x]的前缀和长度为f[x]的后缀相同。因此枚举每个后缀为为j,对这个后缀做kmp,再递推一个令f[i]-j+1≥k且最小的f[i]为last[x](f[i]表示i=x; while(i>=后缀位)i=f[i]得到的所有f[i]),若last[x]-j小于等于(x-j)/2则子串(j,x)合法。讲的很乱,具体看代码。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define INF 0x3fffffff
using namespace std; char s[]; int l,k,next[],last[],ans;
int main(){
//freopen("in.txt","r",stdin);
scanf("%s",s+); l=strlen(s+); scanf("%d",&k); ans=;
inc(i,,l){
next[i]=i-; last[i]=INF;
inc(ii,i+,l){
int j=next[ii-]; while(j>=i&&s[ii]!=s[j+])j=next[j];
if(s[ii]==s[j+])next[ii]=j+;else next[ii]=j;
if(next[ii]-i+<k)last[ii]=INF;else{
last[ii]=min(last[next[ii]],next[ii]);
if(last[ii]-i+<=(ii-i)>>)ans++;//,printf("%d %d\n",i,ii);
}
}
}
printf("%d",ans); return ;
}
20160609