【SPOJ687&POJ3693】Maximum repetition substring（后缀数组）

题意：

n<=1e5

思路：

From http://hzwer.com/6152.html

往后匹配多远 r 用ST表求lcp即可。。。往前 l 就把串反过来再做一下。。

但是有可能求出来的最长串可以前移/后移几位
即开头可以在落在[i−l,i−l+(l+r)mod L]

区间内字典序最小的还要用ST表找rank区间最值

有空需要学习一下结构体写法 一模一样的东西写多次太累了

#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int,int> PII;
typedef vector<int> VI;
#define fi first
#define se second 
#define MP make_pair
#define N   110000
#define MOD 1000000007
#define eps 1e-8 
#define pi acos(-1)
#define oo  1000000000

char ch[N];
int n,i,s[N],sa[N],wa[N],wb[N],wc[N],wd[N],height[N],rank[N],
    ans,mx,ansl,ansr,f[4][N][20],g[3][N],save[N],Log[N],bin[N];


int read()
{ 
   int v=0,f=1;
   char c=getchar();
   while(c<48||57<c) {if(c=='-') f=-1; c=getchar();}
   while(48<=c&&c<=57) v=(v<<3)+v+v+c-48,c=getchar();
   return v*f;
}

bool cmp(int *r,int a,int b,int l)
{
    return r[a]==r[b]&&r[a+l]==r[b+l];
}

void getsa(int *r,int *sa,int n,int m)
{
    int *x=wa,*y=wb,j,p;
    for(i=0;i<n;i++) wc[x[i]=r[i]]++;
    for(i=1;i<m;i++) wc[i]+=wc[i-1];
    for(i=n-1;i>=0;i--) sa[--wc[x[i]]]=i;
    for(j=1,p=1;p<n;j*=2,m=p)
    {
        p=0;
        for(i=n-j;i<n;i++) y[p++]=i;
        for(i=0;i<n;i++)
         if(sa[i]>=j) y[p++]=sa[i]-j;
        for(i=0;i<n;i++) wd[i]=x[y[i]];
        for(i=0;i<m;i++) wc[i]=0;
        for(i=0;i<n;i++) wc[wd[i]]++; 
        for(i=1;i<m;i++) wc[i]+=wc[i-1];
        for(i=n-1;i>=0;i--) sa[--wc[wd[i]]]=y[i];
        swap(x,y);
        p=1; x[sa[0]]=0;
        for(i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}    
    
void getheight(int *r,int *sa,int n)
{
    int i,j,k=0;
    for(i=1;i<=n;i++) rank[sa[i]]=i;
    for(i=0;i<n;height[rank[i++]]=k)
    {
        if(k) k--;
        j=sa[rank[i]-1];
        while(r[i+k]==r[j+k]) k++;
    }
}

void init()
{    
        memset(s,0,sizeof(s));
        memset(sa,0,sizeof(sa));
        memset(wa,0,sizeof(wa));
        memset(wb,0,sizeof(wb));
        memset(wc,0,sizeof(wc));
        memset(wd,0,sizeof(wd));
        memset(height,0,sizeof(height));
        memset(rank,0,sizeof(rank));
}

int lcp(int p,int x,int y)
{
    int t;
    int L=g[p][x];
    int R=g[p][y];
    if(L>R){t=L; L=R; R=t;}
    L++;
    int q=Log[R-L+1];
    return min(f[p][L][q],f[p][R-bin[q]+1][q]);
}

int query(int x,int y)
{
    int q=Log[y-x+1];
    return min(f[3][x][q],f[3][y-bin[q]+1][q]);
}

void solve(int L)
{
    for(int i=0;i+L<n;i+=L)
    if(ch[i]==ch[i+L])
    {
        int r=lcp(1,i,i+L);
    //    printf("1 %d %d %d
",i,i+L,r);
        int l=lcp(2,n-i,n-i-L);
    //    printf("2 %d %d %d
",n-i,n-i-L,l);
        if((l+r)/L+1>mx) {mx=(l+r)/L+1; ans=oo;}
        if((l+r)/L+1==mx)
        {
            int t=query(i-l,i-l+(l+r)%L);
        //    printf("%d %d %d
",i-l,i-l+(l+r)%L,t);
            if(t<ans)
            {
                ans=t;
                ansl=save[t];
                ansr=ansl+mx*L-1;
            }
        }
    }
}

int main()
{
    //freopen("poj3693.in","r",stdin);
    //freopen("poj3693.out","w",stdout); 
    bin[0]=1;
    for(int i=1;i<20;i++) bin[i]=bin[i-1]<<1;
    Log[0]=-1;
    for(int i=1;i<=100000;i++) Log[i]=Log[i/2]+1;
    int cas=0;
    while(scanf("%s",ch))
    {
        if(ch[0]=='#') break;
        cas++;
        printf("Case %d: ",cas);
         
        init();
         n=strlen(ch);
         for(int i=0;i<n;i++) s[i]=ch[i]-'a'+1;
         s[n]=0;
         getsa(s,sa,n+1,100);
         getheight(s,sa,n);
         
         for(int i=0;i<n;i++) f[3][i][0]=rank[i];
         
         int len=Log[n];
         for(int i=2;i<=n;i++) f[1][i][0]=height[i];
         for(int i=1;i<=len;i++)
           for(int j=2;j+bin[i]-1<=n;j++)
            f[1][j][i]=min(f[1][j][i-1],f[1][j+bin[i-1]][i-1]);
        for(int i=0;i<=n;i++) g[1][i]=rank[i];
        for(int i=0;i<=n;i++) save[i]=sa[i];
        
        init();
        for(int i=0;i<n;i++) s[i]=ch[n-1-i]-'a'+1;
         s[n]=0;
         getsa(s,sa,n+1,100);
         getheight(s,sa,n);

         for(int i=2;i<=n;i++) f[2][i][0]=height[i];
         for(int i=1;i<=len;i++)
           for(int j=2;j+bin[i]-1<=n;j++)
           f[2][j][i]=min(f[2][j][i-1],f[2][j+bin[i-1]][i-1]);
        for(int i=0;i<=n;i++) g[2][i]=rank[i];
        
        
    
        for(int i=1;i<=len;i++)
           for(int j=0;j+bin[i]-1<=n-1;j++)
           f[3][j][i]=min(f[3][j][i-1],f[3][j+bin[i-1]][i-1]);

        ansl=ansr=1;
        mx=1; ans=oo;
        for(int i=0;i<n;i++)
         if(ch[i]<ch[ansl]){ans=1; ansl=ansr=i;} 
        for(int i=1;i<=n;i++) solve(i);
        for(int i=ansl;i<=ansr;i++) printf("%c",ch[i]);
        printf("
"); 
    }
    return 0;
}