bzoj3620似乎在梦中见过的样子

bzoj3620似乎在梦中见过的样子

题意：

给出一个字符串，要求求出形如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[15100]; int l,k,next[15100],last[15100],ans;
int main(){
    //freopen("in.txt","r",stdin);
    scanf("%s",s+1); l=strlen(s+1); scanf("%d",&k); ans=0;
    inc(i,1,l){
        next[i]=i-1; last[i]=INF;
        inc(ii,i+1,l){
            int j=next[ii-1]; while(j>=i&&s[ii]!=s[j+1])j=next[j];
            if(s[ii]==s[j+1])next[ii]=j+1;else next[ii]=j;
            if(next[ii]-i+1<k)last[ii]=INF;else{
                last[ii]=min(last[next[ii]],next[ii]);
                if(last[ii]-i+1<=(ii-i)>>1)ans++;//,printf("%d %d\n",i,ii);
            }
        }
    }
    printf("%d",ans); return 0;
}

 

20160609