洛谷P2852 牛奶模式Milk Patterns [USACO06DEC] 字符串

正解:SA/二分+哈希

解题报告:

传送门!

umm像这种子串的问题已经算是比较套路的了,,,?就后缀的公共前缀这样儿的嘛QwQ

所以可以先求个SA

然后现在考虑怎么判断一个长度为d的子串出现了k次?这儿可以直接判断是否有连续的k个hei[i]>=d,因为已经按字典序排序了,所以只要有连续的k个hei[i]>=d就一定有长度大于等于d的子串出现k次

所以二分了check一下就好

然后再说说二分+哈希的做法趴QwQ

就先二分一个长度d,然后把所有长度为d的子串的哈希值求出来,排个序之后判断有没有一个哈希值出现次数>=d

#include<bits/stdc++.h>
using namespace std;
#define il inline
#define gc getchar()
#define ll long long
#define ri register int
#define rb register bool
#define rc register char
#define rp(i,x,y) for(ri i=x;i<=y;++i)
#define my(i,x,y) for(ri i=x;i>=y;--i)

const int N=1e6+,inf=1e9;
int x[N],y[N],sa[N],rk[N],hei[N],t[N],n,K,ch[N],l,r;

il int read()
{
    rc ch=gc;ri x=;rb y=;
    while(ch!='-' && (ch>'' || ch<''))ch=gc;
    if(ch=='-')ch=gc,y=;
    while(ch>='' && ch<='')x=(x<<)+(x<<)+(ch^''),ch=gc;
    return y?x:x;
}
il bool check(ri gd,ri gs,ri k){return y[gd]==y[gs] && y[gd+k]==y[gs+k];}
il void SA()
{
    ri m=N-;
    rp(i,,n)++t[x[i]=ch[i]];
    rp(i,,m)t[i]+=t[i-];
    my(i,n,)sa[t[x[i]]--]=i;
    for(ri k=;k<=n;k<<=)
    {
        ri p=;
        rp(i,,m)t[i]=;rp(i,,n)y[i]=;
        rp(i,n-k+,n)y[++p]=i;rp(i,,n)if(sa[i]>k)y[++p]=sa[i]-k;
        rp(i,,n)++t[x[y[i]]];
        rp(i,,m)t[i]+=t[i-];
        my(i,n,)sa[t[x[y[i]]]--]=y[i];
        swap(x,y);
        x[sa[]]=p=;
        rp(i,,n)x[sa[i]]=check(sa[i],sa[i-],k)?p:++p;
        if(p>=n)break;
        m=p;
    }
    rp(i,,n)rk[sa[i]]=i;
    ri h=;
    rp(i,,n)
    {
        if(h)--h;
        while(ch[i+h]==ch[sa[rk[i]-]+h])++h;
        hei[rk[i]]=h;r=max(r,h);
    }
}
il bool jud(ri x)
{
    ri cnt=;
    rp(i,,n){if(hei[i]<x)cnt=;else ++cnt;if(cnt==K-)return true;}return false;
}

int main()
{
//     freopen("2852.in","r",stdin);freopen("2852.out","w",stdout);
    n=read();K=read();rp(i,,n)ch[i]=read();SA();r=n;
    while(l<r){ri mid=(l+r)>>;if(jud(mid+))l=mid+;else r=mid;}
    printf("%d\n",l);
    return ;
}

啊我麻油写二分+哈希,,,就写了个SA的代码QwQ