【hihoCoder1419】重复旋律4（后缀数组）

题面

hihoCoder

题解

求重复次数最多的连续字串

假如知道了这个连续子串的开始位置
并且知道了它的长度
那么，此时我们就通过$lcp(i,i+len)$来进行比较即可

但是，如果枚举长度和开始的位置的话
尽管$lcp$通过$st$表可以$O(1)$查询
但是这个枚举的复杂度是$O(n^2)$的
很显然，需要更快

所以，枚举了长度$len$之后
我们只需要考虑开始位置为$len$倍数的地方
如果此时有一个重复串的开始位置不在$len$的倍数上
很显然的
$lcp$就会多出一截
所以，我们可以倒推出这个位置
所以，这样枚举复杂度为$O(nlogn)$

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 300000
inline int read()
{
    int x=0,t=1;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return x*t;
}
int SA[MAX],a[MAX],x[MAX],y[MAX],t[MAX];
int Height[MAX],rk[MAX];
int n;
char s[MAX];
int p[20][MAX],lg[MAX];
bool cmp(int i,int j,int k){return y[i]==y[j]&&y[i+k]==y[j+k];}
void GetSA()
{
    int m=30;
    for(int i=1;i<=n;++i)t[x[i]=a[i]]++;
    for(int i=1;i<=m;++i)t[i]+=t[i-1];
    for(int i=n;i>=1;--i)SA[t[x[i]]--]=i;
    for(int k=1;k<=n;k<<=1)
    {
        int p=0;
        for(int i=0;i<=m;++i)y[i]=0;
        for(int i=n-k+1;i<=n;++i)y[++p]=i;
        for(int i=1;i<=n;++i)if(SA[i]>k)y[++p]=SA[i]-k;
        for(int i=0;i<=m;++i)t[i]=0;
        for(int i=1;i<=n;++i)t[x[y[i]]]++;
        for(int i=1;i<=m;++i)t[i]+=t[i-1];
        for(int i=n;i>=1;--i)SA[t[x[y[i]]]--]=y[i];
        swap(x,y);
        x[SA[1]]=p=1;
        for(int i=2;i<=n;++i)x[SA[i]]=cmp(SA[i],SA[i-1],k)?p:++p;
        if(p>=n)break;
        m=p;
    }
    for(int i=1;i<=n;++i)rk[SA[i]]=i;
    for(int i=1,j=0;i<=n;++i)
    {
        if(j)j--;
        while(s[i+j]==s[SA[rk[i]-1]+j])j++;
        Height[rk[i]]=j;
    }
}
void ST_Pre()
{
    memset(p,63,sizeof(p));
    for(int i=1;i<=n;++i)p[0][i]=Height[i];
    for(int j=1;j<=lg[n];++j)
        for(int i=1;i<=n;++i)
            p[j][i]=min(p[j-1][i],p[j-1][i+(1<<(j-1))]);
}
int Query(int l,int r)
{
    int len=r-l+1;
    return min(p[lg[len]][l],p[lg[len]][r-(1<<lg[len])+1]);
}
int lcp(int i,int j)
{
    int l=min(rk[i],rk[j])+1,r=max(rk[i],rk[j]);
    return Query(l,r);
}
int main()
{
    scanf("%s",s+1);
    for(int i=1,l=strlen(s+1);i<=l;++i)a[i]=s[i]-96;
    n=strlen(s+1);
    lg[1]=0;
    for(int i=2;i<=n;++i)lg[i]=lg[i>>1]+1;
    GetSA();
    ST_Pre();
    int ans=0;
    for(int L=1;L<=n;++L)
    {
        for(int i=1;i+L<=n;i+=L)
        {
            int len=lcp(i,i+L);
            ans=max(ans,len/L+1);
            if(i>=L-len%L)
                ans=max(lcp(i-L+len%L,i+len%L)/L+1,ans);
        }
    }
    printf("%d\n",ans);
    return 0;
}