[kmp] bzoj3620: 似乎在梦中见过的样子

bzoj3620: 似乎在梦中见过的样子 :https://www.lydsy.com/JudgeOnline/problem.php?id=3620

求有多少对不相交的子串？
很神的一道题
第一眼并没有想到后缀数组因为不会写
也不知道是不是脑抽了还是刚复习完kmp就想到了kmp
但是不会处理 （因为我比较菜）
看了题解才知道原来 N^2 做法可以过…… 好叭大佬也帮忙算了一波复杂度其实没有N^2那么大完全可行

枚举左端点来求从左端点到结尾每一位的p[]（即next[]）
计算出p后再重新匹配一次自己（相当于找出小串来匹配大串吧）
如果枚举出来的两个子串相交直接跳p 直到不相交相邻
最后统计答案就ok

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
char a[15100];
int p[15100],len,ans=0,k;
void kmp(int st)
{
    for (int i=1;i<=len;i++) p[i]=st-1;
    for (int i=st+1;i<=len;i++)
    {
        int j=p[i-1];
        while (j!=st-1&&a[i]!=a[j+1]) j=p[j];
        if (a[i]==a[j+1]) j++;
        p[i]=j;
    }
    int j=p[st];
    for (int i=st+1;i<=len;i++)
    {
        while (j!=st-1&&a[i]!=a[j+1]) j=p[j];
        if (a[i]==a[j+1]) j++;
        while ((j-st+1)*2>=(i-st+1)) j=p[j];//B>0 ABA的两个A不能重叠(B<0)不能相邻(B=0)
        if ((j-st+1)>=k) ans++;
    }
}
int main()
{
    scanf("%s",a+1);
    scanf("%d",&k);
    len=strlen(a+1);
    for (int i=1;i<=len;i++) kmp(i); //枚举左端点跑kmp 
    printf("%d\n",ans);
    return 0;
}