poj3693 Maximum repetition substring

后缀数组＋RMQ 经典大裸题。。。。

当作复习

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

#define N 100010

int Last_n,Ans,tot,cnt;
int r[N],Rank[N],sa[N],height[N],A[N];
int Use_a[N],Use_b[N],Ws[N],Wv[N];
int f[N][30];
char S[N];

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

void Suf_work(int n,int m)
{
    int *x=Use_a,*y=Use_b,*t;
    for(int i=0;i<m;i++)Ws[i]=0;
    for(int i=0;i<n;i++)Ws[x[i]=r[i]]++;
    for(int i=1;i<m;i++)Ws[i]+=Ws[i-1];
    for(int i=n-1;i>=0;i--)sa[--Ws[x[i]]]=i;
    for(int j=1,p=1;p<n;j<<=1,m=p)
    {
        p=0;
        for(int i=n-j;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++)if(sa[i]>=j)y[p++]=sa[i]-j;
        for(int i=0;i<n;i++) Wv[i]=x[y[i]];
        for(int i=0;i<m;i++) Ws[i]=0;
        for(int i=0;i<n;i++) Ws[Wv[i]]++;
        for(int i=1;i<m;i++) Ws[i]+=Ws[i-1];
        for(int i=n-1;i>=0;i--) sa[--Ws[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],j)?p-1:p++;
    }
}

void Calheight(int n)
{
    int k=0,j;
    for(int i=1;i<=n;i++)Rank[sa[i]]=i;
    for(int i=0;i<n;height[Rank[i++]]=k)
        for(k?k--:0,j=sa[Rank[i]-1];r[i+k]==r[j+k];k++);
}

void RMQ()
{
    for(int i=1;i<=Last_n;i++)
        f[i][0]=height[i];
    for(int j=1;j<=20;j++)
        for(int i=1;i<=Last_n;i++)
        {
            if(i+(1<<j)-1<=Last_n)
                f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);
        }
}

int Get_min(int a,int b)
{
    int x1=Rank[a],x2=Rank[b];
    int l=min(x1,x2),r=max(x1,x2);
    l++;
    int qwer=(int)(log(r-l+1.0)/log(2.0));
    return min(f[l][qwer],f[r-(1<<qwer)+1][qwer]);
}

int main()
{
    while(scanf("%s",S))
    {
        if(S[0]=='#')
            break;
        Last_n=(int)strlen(S);
        if(Last_n==1)
        {
            printf("Case %d: %c\n",++tot,S[0]);
            continue;
        }
        for(int i=0;i<Last_n;i++)
            r[i]=S[i]-'a'+1;
        r[Last_n]=0;
        Suf_work(Last_n+1,40);
        Calheight(Last_n);
        RMQ();
        Ans=0;
        for(int i=1;i<Last_n;i++)
        {
            int now=0;
            while(now+i<Last_n)
            {
                int qwer=Get_min(now,now+i);
                int t=i-qwer%i;
                if(now-t>=0)
                {
                    int qq=Get_min(now-t,now+i-t);
                    if(qq/i>qwer/i)
                        qwer=qq;
                }
                if(qwer/i+1>Ans)
                {
                    Ans=qwer/i+1;
                    cnt=0;
                    A[++cnt]=i;
                }
                else if(qwer/i+1==Ans)
                {
                    A[++cnt]=i;
                }
                now+=i;
            }
        }
        bool flag=false;
        for(int i=1;i<=Last_n;i++)
        {
            if(flag)
                break;
            for(int j=1;j<=cnt;j++)
            {
                int l=A[j];
                if(Get_min(sa[i],sa[i]+l)>=(Ans-1)*l)
                {
                    printf("Case %d: ",++tot);
                    for(int k=sa[i];k<sa[i]+Ans*l;k++)
                    {
                        printf("%c",S[k]);
                    }
                    printf("\n");
                    flag=true;
                    break;
                }
            }
        }
    }
    return 0;
}