Bzoj 3620---似乎在梦中见过的样子(KMP)

题目背景第二行madoka打错了？？？

KMP暴力啦~~~~

我们使用 KMP 的 P 数组来找这个前缀，直接从 P[r] 向前找最坏情况会导致变成 O(n^3) ，所以我们应该优化一下：

P 数组如果视为 Father 数组，那么它是一棵树。我们求的就是一个节点到根的路径上有没有区间 [k, (r-l+1)/2) 的数。

我们求 P 的时候是从根向叶子求的，所以我们可以同时维护根到某个节点 j 的所有数中，大于等于 k 的数中最小的一个。

然后我们对于每一个右端点，只要判断这个点到根的路径中大于等于 k 的数中最小的一个是否小于 (r - l + 1) / 2 就可以了。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int INF=999999999;
int ans=0,p[1500005],num[1500005],n,k;
char st[1500005];
int main()
{
    gets(st+1);
    scanf("%d",&k);
    int n=strlen(st+1);
    for (int i=1; i<=n; i++){
        int j=0;
        p[1]=0; num[0]=INF;
        for (int t=2; t<=n-i+1; t++){
            while (j>0 && st[i+j]!=st[t+i-1]) j=p[j];
            if (st[i+j]==st[i+t-1]) j++;
            p[t]=j;
            if (j<k) num[j]=INF; else num[j]=min(j,num[p[j]]);
            if ((num[j]<<1)<t) ans++;
        }        
    }
    printf("%d
",ans);
    return 0;
}

miao~~~