POJ 3693(后缀数组)

(http://acm.hust.edu.cn/vjudge/contest/view.action?cid=105595#problem/H)
题意:给定字符串s,寻找其子串T,使得T循环节数最多
(len(T)/len(T的最小循环节)最大)。
解法:还是枚举循环节长度,那么T的第一个循环节一定包含0,T,2T…其中的一个。时间复杂度就是O(nlgn).
同样套路的题还有UVA 10829.都是这种枚举L,计算lcp(i,i+L)的套路。

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <vector>

using namespace std;

#define ll long long
#define inf 1e18
#define clr(x,y) memset(x,y,sizeof x)
#define FOR(i,a,b) \
    for(i=a;a<b?i<=b:i>=b;a<b?i++:i--)

const int maxn = 100000*2+20;

struct suffix_array
{
    char s[maxn];
    int sa[maxn] , t[maxn] , t2[maxn] , c[maxn];
    int n;
    void build_sa(int m)
    {
        int i,*x=t,*y=t2;
        FOR(i,0,m-1) c[i]=0;
        FOR(i,0,n-1) c[x[i]=s[i]]++;
        FOR(i,1,m-1) c[i]+=c[i-1];
        FOR(i,n-1,0) sa[--c[x[i]]]=i;
        for(int k=1;k<=n;k<<=1)
        {
            int p=0;
            FOR(i,n-k,n-1) y[p++]=i;
            FOR(i,0,n-1) if(sa[i]>=k) y[p++]=sa[i]-k;

            FOR(i,0,m-1) c[i]=0;
            FOR(i,0,n-1) c[x[y[i]]]++;

            FOR(i,0,m-1) c[i]+=c[i-1];
            FOR(i,n-1,0) sa[--c[x[y[i]]]]=y[i];

            swap(x,y);
            p=1;
            x[sa[0]]=0;
            FOR(i,1,n-1)
                x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&
                    y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
            m=p;
        }
        //printf("finish build\n");
    }
    int Rank[maxn],height[maxn];
    void getHeight()
    {
        int k=0;
        for(int i=0;i<n;i++) Rank[sa[i]]=i;
        for(int i=0;i<n;i++){
            if(k) k--;
            int j=sa[Rank[i]-1];
            while(s[i+k]==s[j+k]) k++;
            height[Rank[i]]=k;
        }
    }
    int d[maxn][30],flog[maxn];
    void RMQ_init()
    {
        for(int i=0;i<n;i++) d[i][0]=height[i];
        flog[0]=-1;
        for(int i=1;i<=n;i++) flog[i]=flog[i>>1]+1;
        for(int j=1;(1<<j)<=n;j++)
            for(int i=0;i+(1<<j)<=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=flog[R-L+1];
        return min(d[L][k],d[R-(1<<k)+1][k]);
    }
    int lcp(int j,int k)
    {
        if(j==k) return n-k;
        if(Rank[j]>Rank[k]) swap(j,k);
        return RMQ(Rank[j]+1,Rank[k]);
    }
    void init(char *ss)
    {
        strcpy(s,ss);
        n = strlen(s)+1;
        //printf("in\n");
        build_sa(255);
        getHeight();
        RMQ_init();
    }
}su,su2;

int d[maxn][30],flog[maxn];
void RMQ_init(int *A,int n)
{
    for(int i=0;i<n;i++) d[i][0]=A[i];
    flog[0]=-1;
    for(int i=1;i<=n;i++) flog[i]=flog[i>>1]+1;
    for(int j=1;(1<<j)<=n;j++)
        for(int i=0;i+(1<<j)<=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=flog[R-L+1];
    return min(d[L][k],d[R-(1<<k)+1][k]);
}

void Rev(char *s)
{
    int l=0,r=strlen(s)-1;
    while(l<r)
        swap(s[l++],s[r--]);
}

char s[maxn],s2[maxn];

int main()
{
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    int CASE=0;
    while(~scanf("%s",s))
    {
        if(strcmp(s,"#")==0)
            break;
        int len = strlen(s);
//      printf("len %d\n",len);
        su.init(s);
        strcpy(s2,s);
        Rev(s2);
        su2.init(s2);
        int ans=0,ansr=-1,ansl=-1;
        RMQ_init(su.Rank,len);
        for(int l=1;l<=len;l++)
        {
            for(int i=0;i+l<len;i+=l) if( s[i] == s[i+l] )
            {
                //printf("i %d l %d\n",i,l);
                int lcp1 = su.lcp(i,i+l) , lcp2 = su2.lcp(len-i-1,len-i-l-1);
//              if(i==0&&l==1)printf("%d %d %d\n",len-i-1,len-i-l-1,su2.lcp(len-i-1,len-i-l-1));
                if( lcp1+lcp2-1>=l )
                {
                    int L =  lcp1 + lcp2  - 1;
                    int r = L/l +1;
                    int pos = su.sa[ RMQ(i - lcp2 + 1 , i-lcp2+1+L%l) ];
                    if( r > ansr || (r==ansr&&su.Rank[pos] < su.Rank[ans]) )
                    {
                        ans = pos;
                        ansr = r;
                        ansl = r*l;
//                      printf("i %d lcp1 %d lcp2 %d r %d pos %d\n",i,lcp1,lcp2,r,pos);
                    }
                }
            }
        }
        printf("Case %d: ",++CASE);
        if( ansr==-1 )
        {
            char a='z';
            for(int i=0;i<len;i++)
                a=min(s[i],a);
            printf("%c\n",a);
            continue;
        }

        for(int i=0;i<ansl;i++)
            printf("%c",s[i+ans]);
        printf("\n");
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值