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

Posted 2020-08-31 Candy?

和我签订契约，成为魔法少女吧

---

题意：求所有形似于A+B+A 的子串的数量 , 且len(A)>=k,len(B)>=1 

位置不同其他性质相同的子串算不同子串,位置相同但拆分不同的子串算同一子串

---

 

竟然是暴力........

枚举从哪里开始，和上题一样了

只不过本题$l$确定后一个$r$只能贡献一次，所以向前找第一个$2*j \le i-1$的位置判断$j \ge k$就行了

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N=15005,MOD=1e9+7;
int n,k,ans;
char s[N];
int fail[N],sum[N];
void KMP(char s[],int n){
    fail[1]=0;
    for(int i=2;i<=n;i++){
        int j=fail[i-1];
        while(j&&s[i]!=s[j+1]) j=fail[j];
        fail[i]=s[i]==s[j+1]?j+1:0;
    }
    int j=0;
    for(int i=2;i<=n;i++){
        while(j&&s[i]!=s[j+1]) j=fail[j];
        if(s[i]==s[j+1]) j++;
        while((j<<1)>i-1) j=fail[j];
        ans+=j>=k;
    }
}
int main(){
    freopen("in","r",stdin);
    scanf("%s%d",s+1,&k);
    n=strlen(s+1);
    int _=n-(k<<1);
    for(int i=0;i<_;i++) KMP(s+i,n-i);
    printf("%d",ans);
}