SPOJ687 Repeats（重复次数最多的连续子串）

Repeats

hihoCoder 1419 后缀数组四·重复旋律4(重复次数最多的连续子串)

#include<bits/stdc++.h>
using namespace std;

// sa[i]: 排名是i位的是第几个后缀
// rk[i]: 第i个后缀的排名是多少
// height[i]: sa[i]与sa[i-1]
const int N=50010;
char s[N];
int rk[N],sa[N],cnt[N],height[N];
int x[N],y[N];
int n,m;
void rsort()// x[i] 第一关键字 y[i] 第二关键字 基数排序
{
    for(int i=1;i<=m;i++) cnt[i]=0;
    for(int i=1;i<=n;i++) cnt[x[i]]++;
    for(int i=1;i<=m;i++) cnt[i]+=cnt[i-1];
    for(int i=n;i;i--) sa[cnt[x[y[i]]]--]=y[i];
}
void SA()
{
    n=strlen(s+1);
    m=300;
    for(int i=1;i<=n;i++) x[i]=s[i],y[i]=i;
    rsort();
    for(int k=1;k<=n;k<<=1)
    {
        int p=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;
        rsort();swap(x,y);
        
        x[sa[1]]=1,p=1;
        for(int i=2;i<=n;i++)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p:++p);
        if(p==n) break;
        m=p;
    }
    for(int i=1;i<=n;i++) rk[sa[i]]=i;
    // 求height
    
    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;
    }
}
int st[N][22],lg[N];
void init()
{
    for(int i=1;i<=n;i++) st[i][0]=height[i];
    for(int k=1;k<=lg[n];k++)
        for(int i=1;i+(1<<k)-1<=n;i++)
            st[i][k]=min(st[i][k-1],st[i+(1<<k-1)][k-1]);
}
int lcp(int a,int b)
{
    int l=min(rk[a],rk[b])+1;
    int r=max(rk[a],rk[b]);
    int k=lg[r-l+1];
    return min(st[l][k],st[r-(1<<k)+1][k]);
}
int solve()
{
    int ans=1;
    for(int l=1;l<=n;l++)
        for(int i=1;i+l<=n;i+=l)
        {
            int r=lcp(i,i+l);
            ans=max(ans,r/l+1);
            if(i>=l-r%l) ans=max(ans,lcp(i-l+r%l,i+r%l)/l+1);
        }
    return ans;
}
int main()
{
    for(int i=2;i<=50000;i++) lg[i]=lg[i>>1]+1;
    int Tc;
    scanf("%d",&Tc);
    while(Tc--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++)scanf(" %c",&s[i]);s[n+1]='\0';
        SA();
        init();
        printf("%d\n",solve());
    }
    return 0;
}