SPOJ REPEATS 后缀数组

题意

给一个字符串,要求寻找一个循环子串,使得循环子串的循环次数最大.

题解

感觉这道题确实有不小的难度,尤其是对于刚刚接触后缀数组的来说,确实很难处理.既然很难处理,那我们就可以先用最暴力的方法解决,如果暴力有哪里难以解决的话,再去考虑后缀数组优化.
既然是要找一个循环子串,那么我们可以枚举长度(很常见的字符串套路).从0开始,计算从某一位开始向后组成的最大循环次数.可以明显看出,这个算法是O(N^2)的.解决这道题是不够快的.很明显,枚举长度是不能省的,所以要从计算循环次数入手,争取将其优化成O(logn).
优化我们可以先去观察height数组,看看height数组如何去表示循环长度.我们可以发现从a开始到b的串向后所能组成的循环子串长度最长为rank[a]的height到rank[b]的height之间的最小值.具体为什么是这个规律,网上的题解都么有讲清.我这里抛砖引玉,谈一下自己的见解吧.
对于一个串来说,如果想要与后面的串产生循环关系,就必定在rank[a]的height到rank[b]的height之间.因为height对应的相同前缀从另一个角度看就是一对循环,因此也可认为height解决的就是循环的问题.
总之,知道了上述结论,我们就可以用RMQ去解决区间最小值的问题.查询循环串的长度,得到长度以后+1,因为height是记录的循环长度,除了相匹配的循环子串,本身的循环子串也有1个,因此需要+1.另外的话,还需要判断剩余的子串是否可以和前面的串组成一个循环串,这个问题在大多数题解里都讲得很清楚,这里就不再详细阐述了.

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#include<stack>
#include<string>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define LL long long
#define INF 0x3f3f3f3f
#define MAXN 50050
#define MOD 1000000007
#define EPS 1e-3
using namespace std;

char s[MAXN];
int sa[MAXN],t[MAXN],t2[MAXN],c[MAXN],rk[MAXN],height[MAXN],n;
int d[MAXN][20];

void build_sa(int m) {
    int i,*x=t,*y=t2;
    UP(i,0,m) c[i]=0;
    UP(i,0,n) c[x[i]=s[i]]++;
    UP(i,1,m) c[i]+=c[i-1];
    DOWN(i,n,0) sa[--c[x[i]]]=i;
    for(int k=1; k<=n; k<<=1) {
        int p=0;
        UP(i,n-k,n) y[p++]=i;
        UP(i,0,n) if(sa[i]>=k) y[p++]=sa[i]-k;
        UP(i,0,m) c[i]=0;
        UP(i,0,n) c[x[y[i]]]++;
        UP(i,1,m) c[i]+=c[i-1];
        DOWN(i,n,0) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;
        x[sa[0]]=0;
        UP(i,1,n) {
            x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
        }
        if(p>=n) break;
        m=p;
    }
}

void getHeight(){
    int i,j,k=0;
    UP(i,0,n) rk[sa[i]]=i;
    height[0]=0;
    UP(i,0,n-1){
        if(k)
            k--;
        int j=sa[rk[i]-1];
        W(s[i+k]==s[j+k])
            k++;
        height[rk[i]]=k;
    }
}

void rmq_init(){
    UP(i,0,n) d[i][0]=height[i];
    for(int j=1;(1<<j)<=n;j++){
        for(int i=0;i+(1<<j)-1<n;i++){
            d[i][j]=min(d[i][j-1],d[i+(1<<(j-1))][j-1]);
        }
    }
}

int rmq(int l,int r){
    int k=0;
    W((1<<(k+1))<=r-l+1) k++;
    return min(d[l][k],d[r-(1<<k)+1][k]);
}

int mxSub(int l,int r){
    l=rk[l],r=rk[r];
    if(l>r) swap(l,r);
    return rmq(l+1,r);
}

int main() {
    int t;
    scanf("%d",&t);
    W(t--) {
        scanf("%d",&n);
        UP(i,0,n) {
            getchar();
            scanf("%c",&s[i]);
        }
//        cout<<s<<endl;
        n++;
        build_sa(128);
        getHeight();
        rmq_init();
        int ans=0;
        UP(i,1,n){
            for(int j=0;j<(n-i);j+=i){
                int s=mxSub(j,j+i);
                int k=j-(i-s%i);
                int as=s/i+1;
                if(k>=0&&mxSub(k,k+i)>=i)
                    as++;
                ans=max(ans,as);
//                cout<<i<<" "<<j<<" "<<ans<<" "<<s<<endl;
            }
        }
        printf("%d\n",ans);
    }
}
/*
5
9
a b a a b a a b a

55
12
a b a a b a a b a a b a
*/