【后缀数组求重复次数最多的连续重复子串】SPOJ687 POJ3693

spoj 687

#include <map>
#include <set>
#include <list>
#include <queue>
#include <deque>
#include <stack>
#include <string>
#include <cstdio>
#include <math.h>
#include <iomanip>
#include <cstdlib>
#include <limits.h>
#include <string.h>
#include <iostream>
#include <fstream>
#include <algorithm>
using namespace std;

#define LL long long
#define pii pair<int ,int>
#define bug cout<<"here!!"<<endl
#define PI acos(-1.0)
#define FRE freopen("input.txt","r",stdin)
#define FF  freopen("output.txt","w",stdout)
#define eps 1e-8
#define MIN INT_MIN
#define inf 1<<30
#define N 1000010
#define maxn 50001

int wa[maxn],wb[maxn],wv[maxn],wss[maxn];

int cmp(int *r,int a,int b,int l)
{return r[a]==r[b] && r[a+l]==r[b+l];}
void da(int *r,int *sa,int n,int m){//n要加1
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=0;i<m;i++) wss[i]=0;
     for(i=0;i<n;i++) wss[x[i]=r[i]]++;
     for(i=1;i<m;i++) wss[i]+=wss[i-1];
     for(i=n-1;i>=0;i--) sa[--wss[x[i]]]=i;
     for(j=1,p=1;p<n;j*=2,m=p){
         for(p=0,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++) wv[i]=x[y[i]];
         for(i=0;i<m;i++) wss[i]=0;
         for(i=0;i<n;i++) wss[wv[i]]++;
         for(i=1;i<m;i++) wss[i]+=wss[i-1];
         for(i=n-1;i>=0;i--) sa[--wss[wv[i]]]=y[i];
         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
         x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
     return;
}
int rank[maxn],height[maxn];
void calheight(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)
     for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
     return;
}
int RMQ[maxn];
int mm[maxn];
int best[20][maxn];
void initRMQ(int n)
{
     int i,j,a,b;
     for(mm[0]=-1,i=1;i<=n;i++)
     mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
     for(i=1;i<=n;i++) best[0][i]=i;
     for(i=1;i<=mm[n];i++)
     for(j=1;j<=n+1-(1<<i);j++)
     {
       a=best[i-1][j];
       b=best[i-1][j+(1<<(i-1))];
       if(RMQ[a]<RMQ[b]) best[i][j]=a;
       else best[i][j]=b;
     }
     return;
}
int askRMQ(int a,int b)
{
    int t;
    t=mm[b-a+1];b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return RMQ[a]<RMQ[b]?a:b;
}
int lcp(int a,int b)
{
    int t;
    a=rank[a];b=rank[b];
    if(a>b) {t=a;a=b;b=t;}
    return(height[askRMQ(a+1,b)]);
}

int r[maxn],sa[maxn];
int main(){
    int t;
    scanf("%d",&t);
    while(t--){
        int n;
        scanf("%d",&n);getchar();
        int i,j;
        char c;
        for(i=0;i<n;i++){
            scanf("%c\n",&c);
            r[i] = c;
        }
        r[n] = 0;
        da(r,sa,n+1,128);
        calheight(r,sa,n);
        for(i=1;i<=n;i++){
            RMQ[i] = height[i];
        }
        initRMQ(n);
        int ans=0;
        for(i=1;i<n;i++){
            for(j=0;j+i<n;j+=i){
                int k = lcp(j,j+i);
                int cnt = k/i+1;
                int jj = j- (i - k%i);
                if(jj>=0 && lcp(jj,jj+i)>=(i-k%i))cnt++;
                if(cnt>ans)ans = cnt;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

POJ 3693

#define maxn 100010

int wa[maxn],wb[maxn],wv[maxn],wss[maxn];

int cmp(int *r,int a,int b,int l)
{return r[a]==r[b] && r[a+l]==r[b+l];}
void da(int *r,int *sa,int n,int m){//n要加1
     int i,j,p,*x=wa,*y=wb,*t;
     for(i=0;i<m;i++) wss[i]=0;
     for(i=0;i<n;i++) wss[x[i]=r[i]]++;
     for(i=1;i<m;i++) wss[i]+=wss[i-1];
     for(i=n-1;i>=0;i--) sa[--wss[x[i]]]=i;
     for(j=1,p=1;p<n;j*=2,m=p){
         for(p=0,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++) wv[i]=x[y[i]];
         for(i=0;i<m;i++) wss[i]=0;
         for(i=0;i<n;i++) wss[wv[i]]++;
         for(i=1;i<m;i++) wss[i]+=wss[i-1];
         for(i=n-1;i>=0;i--) sa[--wss[wv[i]]]=y[i];
         for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
         x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
     return;
}
int rank[maxn],height[maxn];
void calheight(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)
     for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
     return;
}
int RMQ[maxn];
int mm[maxn];
int best[20][maxn];
void initRMQ(int n)
{
     int i,j,a,b;
     for(mm[0]=-1,i=1;i<=n;i++)
     mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
     for(i=1;i<=n;i++) best[0][i]=i;
     for(i=1;i<=mm[n];i++)
     for(j=1;j<=n+1-(1<<i);j++)
     {
       a=best[i-1][j];
       b=best[i-1][j+(1<<(i-1))];
       if(RMQ[a]<RMQ[b]) best[i][j]=a;
       else best[i][j]=b;
     }
     return;
}
int askRMQ(int a,int b)
{
    int t;
    t=mm[b-a+1];b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return RMQ[a]<RMQ[b]?a:b;
}
int lcp(int a,int b)
{
    int t;
    a=rank[a];b=rank[b];
    if(a>b) {t=a;a=b;b=t;}
    return(height[askRMQ(a+1,b)]);
}

int r[maxn],sa[maxn];
char str[maxn];
int main(){FRE;
    int t = 1;
    while(scanf("%s",str)){
        if(str[0] == '#')break;
        int n = strlen(str);
        int i,j;
        char c;
        for(i=0;i<n;i++){
            r[i] = str[i];
        }
        r[n] = 0;
        da(r,sa,n+1,128);
        calheight(r,sa,n);
        for(i=1;i<=n;i++){
            RMQ[i] = height[i];
        }
        initRMQ(n);
        int ans=0;
        int len[maxn];
        int num = 0;
        for(i=1;i<n;i++){
            for(j=0;j+i<n;j+=i){
                int k = lcp(j,j+i);
                int cnt = k/i+1;
                int jj = j- (i - k%i);
                if(jj>=0 && lcp(jj,jj+i)>=(i-k%i)){
                    cnt++;
                }
                if(cnt == ans){
                    len[num++] = i;
                } else if(cnt>ans){
                    ans = cnt;
                    num = 0;
                    len[num++] = i;
                }
            }
        }
        int st,l;
        for(i=1;i<=n;i++){//暴力枚举每个sa后缀
            int beg = sa[i];
            for(j=0;j<num;j++){
                int ll = len[j];//每一段长度
                if(beg+ll<n && lcp(beg,beg+ll)>=(ans-1)*ll){
                    st = beg;
                    l = ans*ll;
                    i = n;
                    break;
                }
            }
        }
        printf("Case %d: ",t++);
        for(i=0;i<l;i++){
            printf("%c",str[st+i]);
        }
        puts("");
    }
    return 0;
}