【HDOJ5558】Alice's Classified Message（后缀数组）

题意：给定一个长度为n的下标从0开始的小写字母字符串，每次对于当前的i寻找一个j使得后缀i与后缀j的lcp最大，(j<i)若lcp相同则取较小的

若lcp为0则输出0与当前字符，i自增1，否则输出lcp的值与j，i自增lcp的值，以上过程重复直到i>=n

要求模拟这个过程

n<=1e5,sigma n<=2e6

思路：显然后缀的lcp要用到后缀数组

考虑对于每一个i都直接枚举j不可做，对于rank而言可以预处理出每一段连续的height大于0的起始位置和结束位置，在段中往左右两侧暴力枚举

满足要求的条件为：sa[j]<i，且连续的height的min的值不减

这个暴力扫描的复杂度应该假了（真的吗），但数据卡不掉……

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
typedef long long ll;
using namespace std;
#define N   110000 
#define oo  10000000
#define MOD 100000073


char ch[N];
int n,i,s[N],sa[N],wa[N],wb[N],wc[N],wd[N],height[N],Rank[N],pre[N],nxt[N];

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++;
        if(j>n) break; 
    } 
}

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 main()
{ 
    int cas; 
    scanf("%d",&cas);
    for(int v=1;v<=cas;v++)
    {
        init();
        printf("Case #%d:\n",v);
        scanf("%s",ch);
        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=1;i<=n;i++)
         if(!height[i]) pre[i]=i;
          else pre[i]=pre[i-1];
        for(int i=n;i>=1;i--)
         if(height[i+1]==0||i==n) nxt[i]=i;
          else nxt[i]=nxt[i+1];
        int i=0;
        while(i<n)
        {
            int now=Rank[i];
            int k=0; 
            int t=i;
            int mn=height[now];
            for(int j=now-1;j>=pre[now];j--)
            {
                mn=min(mn,height[j+1]);
                if(mn<k) break;
                if(sa[j]<i)
                {
                    if(mn>k||mn==k&&sa[j]<t)
                    {
                        k=mn; t=sa[j];
                    }
                }
            }
            if(now+1<=nxt[now]) mn=height[now+1];
            for(int j=now+1;j<=nxt[now];j++)
            {
                mn=min(mn,height[j]);
                if(mn<k) break;
                if(sa[j]<i)
                {
                    if(mn>k||mn==k&&sa[j]<t)
                    {
                        k=mn; t=sa[j];
                    }
                }
            }
            if(!k)
            {
                printf("-1 %d\n",ch[i]);
                i++;
            }
             else
             {
                 printf("%d %d\n",k,t);
                 i+=k;
             }
        }        
    }
    return 0;
}