[USACO06DEC] Milk Patterns

后缀数组求最长k次重复子串。

poj 3261 传送门

洛谷 P2852 传送门

求完height数组之后倍增答案。

check的时候枚举height数组。

如果连续cnt个height都大于等于某长度ans，且cnt不小于题目要求的k次重复，则ans为一个合法答案。

如果枚举过程中遇到了小于ans的height，则需要重置计数器cnt（初值为1）。

注意考虑每个数组分别需要开多大。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

int n,m,ans;
int s[20005];
int sa[20005],rk[20005];
int h[1000005],tr[1000005];

int cmp(int x,int y,int k)
{
    if(x+k>n||y+k>n)return 0;
    return rk[x]==rk[y]&&rk[x+k]==rk[y+k];
}

void cal()
{
    int i,cnt;
    for(i=1;i<=n;i++)h[s[i]]++;
    for(i=1,cnt=0;i<=1000000;i++)if(h[i])tr[i]=++cnt;
    for(i=1;i<=1000000;i++)h[i]+=h[i-1];
    for(i=1;i<=n;i++)rk[i]=tr[s[i]],sa[h[s[i]]--]=i;
    for(int k=1;cnt!=n;k<<=1)
    {
        for(i=1;i<=n;i++)h[i]=0;
        for(i=1;i<=n;i++)h[rk[i]]++;
        for(i=1;i<=n;i++)h[i]+=h[i-1];
        for(i=n;i;i--)if(sa[i]>k)tr[sa[i]-k]=h[rk[sa[i]-k]]--;
        for(i=1;i<=k;i++)tr[n-i+1]=h[rk[n-i+1]]--;
        for(i=1;i<=n;i++)sa[tr[i]]=i;
        for(i=1,cnt=0;i<=n;i++)tr[sa[i]]=cmp(sa[i],sa[i-1],k)?cnt:++cnt;
        for(i=1;i<=n;i++)rk[i]=tr[i];
    }
    for(i=1;i<=n;i++)h[i]=0;
    for(i=1;i<=n;i++)
    {
        if(rk[i]==1)continue;
        for(int j=max(1,h[rk[i-1]]-1);;j++)
        {
            if(s[i+j-1]==s[sa[rk[i]-1]+j-1])h[rk[i]]=j;
            else break;
        }
    }
}

int check(int k)
{
    if(k>n)return 0;
    int cnt=1;
    for(int i=2;i<=n;i++)
    {
        if(h[i]>=k)cnt++;
        else cnt=1;
        if(cnt>=m)return 1;
    }
    return 0;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&s[i]);
    cal();
    for(int i=20;i>=0;i--)
        if(check(ans|(1<<i)))ans|=(1<<i);
    printf("%d",ans);
    return 0;
}