POJ2774 Long Long Message

思路

后缀数组的水题
把两个字符串拼起来，找到最大的height满足sa[i-1]和sa[i]不在同一子串中即可

代码

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int MAXN = 100100*2;
struct Node{
    int pos,r[2];
}x[MAXN],midx[MAXN];
int barrel[MAXN],ranks[MAXN],sa[MAXN],height[MAXN],n,lena,lenb;
char s[MAXN],t[MAXN];
int c_sort(int n,int lim){
    for(int i=0;i<2;i++){
        memset(barrel,0,sizeof(barrel));
        for(int j=1;j<=n;j++)
            barrel[x[j].r[i]]++;
        for(int j=1;j<=lim;j++)
            barrel[j]+=barrel[j-1];
        for(int j=n;j>=1;j--)
            midx[barrel[x[j].r[i]]--]=x[j];
        for(int j=1;j<=n;j++)
            x[j]=midx[j];
    }
    int cnt=1;
    ranks[x[1].pos]=1;
    for(int i=2;i<=n;i++)
        if(x[i].r[0]==x[i-1].r[0]&&x[i].r[1]==x[i-1].r[1])
            ranks[x[i].pos]=cnt;
        else
            ranks[x[i].pos]=++cnt;
    return cnt;
}
void cal_sa(int n){
    for(int i=1;i<=n;i++)
        x[i]=(Node){i,s[i],0};
    int cnt=c_sort(n,2000);
    for(int i=1;cnt<n;i<<=1){
        for(int j=1;j<=n;j++)
            x[j]=(Node){j,(i+j<=n)?ranks[i+j]:0,ranks[j]};
        cnt=c_sort(n,cnt);
    }
    for(int j=1;j<=n;j++)
        sa[ranks[j]]=j;
    for(int i=1,j=0,k;i<=n;height[ranks[i++]]=j)
        for(j?j--:0,k=sa[ranks[i]-1];s[i+j]==s[j+k];j++);
}
int main(){
    scanf("%s",t+1);
    lena=strlen(t+1);
    for(int i=1;i<=lena;i++)
        s[i]=t[i];
    n+=lena;
    s[++n]='#';
    scanf("%s",t+1);
    lenb=strlen(t+1);
    for(int i=1;i<=lenb;i++)
        s[n+i]=t[i];
    n+=lenb;
    cal_sa(n);
    int ans=0;
    for(int i=2;i<=n;i++){
        if((sa[i-1]<=lena&&sa[i]>lena+1)||(sa[i-1]>lena+1&&sa[i]<=lena))
            ans=max(ans,height[i]);
    }
    printf("%d
",ans);
    return 0;
}