POJ 3693 Maximum Repetition Substring <后缀数组 + RMQ>

题目：传送门。

分析：

求连续重复次数最多的子串，若有多个，输出字典序最小的。感觉有点难度了，看了解题报告做的。

代码：

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

const int MAXN=100005;
const int INF=0x3f3f3f3f;
int wa[MAXN],wb[MAXN],wv[MAXN],w[MAXN],height[MAXN];
int sa[MAXN];
int *rank;
char str[MAXN];
int data[MAXN];
int dp[MAXN][17];

bool cmp(int *r,int a,int b,int l){  
    return r[a]==r[b]&&r[a+l]==r[b+l];  
}  
  
void da(int n,int m){  
    int *x=wa,*y=wb,*t,p=1;  
    for(int i=0;i<m;++i) w[i]=0;  
    for(int i=0;i<n;++i) ++w[x[i]=data[i]];  
    for(int i=1;i<m;++i) w[i]+=w[i-1];  
    for(int i=n-1;i>=0;--i) sa[--w[x[i]]]=i;  
    for(int len=1;p<n;len<<=1,m=p){  
        p=0;  
        for(int i=n-len;i<n;++i) y[p++]=i;  
        for(int i=0;i<n;++i) if(sa[i]>=len) y[p++]=sa[i]-len;  
        for(int i=0;i<n;++i) wv[i]=x[y[i]];  
        for(int i=0;i<m;++i) w[i]=0;  
        for(int i=0;i<n;++i) ++w[wv[i]];  
        for(int i=1;i<m;++i) w[i]+=w[i-1];  
        for(int i=n-1;i>=0;--i) sa[--w[wv[i]]]=y[i];  
        t=x; x=y; y=t; p=1; x[sa[0]]=0;  
        for(int i=1;i<n;++i) x[sa[i]]=cmp(y,sa[i-1],sa[i],len)?p-1:p++;  
    }  
    rank=x;  
}  
  
void calHeight(int n){  
    for(int i=0,k=0,j;i<n;height[rank[i++]]=k)  
        for(k?--k:k,j=sa[rank[i]-1];data[i+k]==data[j+k];++k);  
}

void initRMQ(int n){
	int l=(int)(log(n*1.0)/log(2.0));
	for(int i=1;i<=n;++i) dp[i][0]=height[i];
	for(int j=1;j<=l;++j){
		for(int i=1;i+(1<<j)-1<=n;++i){
			dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
		}
	}
}

int LCP_Inquiry(int a,int b){
	int i=rank[a],j=rank[b];
	if(i>j) swap(i,j); ++i;
	int t=(int)((log(j-i+1.0)/log(2.0)));
	return min(dp[i][t],dp[j-(1<<t)+1][t]);
}

int main(){
	int cas=0;
	while(scanf("%s",str) && str[0]!='#'){
		int len=strlen(str),m=0;
		for(int i=0;i<len;++i){
			data[i]=str[i]-'a'+1;
			if(m<data[i]) m=data[i];
		}
		data[len]=0;
		da(len+1,m+2);
		calHeight(len);
		initRMQ(len);
		int max_t=1,star=sa[1],max_len=1;;
		for(int l=1;l<=len/2;++l){
			for(int i=0;i+l<len;i+=l){
				if(str[i]!=str[i+l]) continue;
				int t=i+l+LCP_Inquiry(i,i+l)-1;
				int j=i;
				while(j>=0&&j>i-l&&str[j]==str[j+l]){
					int rp=(t-j+1)/l;
					if(rp>max_t || (rp==max_t && rank[j]<rank[star])){
						star=j;
						max_len=l;
						max_t=rp;
					}
					--j;
				}
			}
		}
		str[star+max_t*max_len]='\0';
		printf("Case %d: ",++cas);
		printf("%s\n",&str[star]);
	}
	return 0;
}