poj 2774 后缀数组

题目大意:求最长公共前缀
思路:这种公共字串问题一般都是将后面的串插到前面上去,然后搞事情Orz
这题就这样咯,只要两个不在一个串内即一个为A串,一个为B串,答案为满足条件的height值的最大值
PS:小技巧:用一个比较大的值插入到中间,根据sa值判断是否在一个串中
还有数据只有一组
最重要的数组要开到2e5以上
TLE和RE的血泪教训啊o_O

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2e5+10;
int n,p,q,k,lena,lenb;
char A[N],B[N];
int a[N],v[N],h[N];
int sa[2][N],rk[2][N];
void calsa(int sa[N],int rk[N],int SA[N],int RK[N])
{
    for(int i=1;i<=n;i++)v[rk[sa[i]]]=i;
    for(int i=n;i;i--)
        if(sa[i]>k)
            SA[v[rk[sa[i]-k]]--]=sa[i]-k;
    for(int i=n-k+1;i<=n;i++)SA[v[rk[i]]--]=i;
    for(int i=1;i<=n;i++)
        RK[SA[i]]=RK[SA[i-1]]+(rk[SA[i-1]]!=rk[SA[i]]||rk[SA[i]+k]!=rk[SA[i-1]+k]);
}
void getsa()
{
    memset(v,0,sizeof(v));
    p=0,q=1;
    for(int i=1;i<=n;i++)v[a[i]]++;
    for(int i=1;i<=30;i++)v[i]+=v[i-1];
    for(int i=1;i<=n;i++)
        sa[p][v[a[i]]--]=i;
    for(int i=1;i<=n;i++)
        rk[p][sa[p][i]]=rk[p][sa[p][i-1]]+(a[sa[p][i-1]]!=a[sa[p][i]]);
    for(k=1;k<n;k<<=1,swap(p,q))
        calsa(sa[p],rk[p],sa[q],rk[q]);
}
void geth()
{
    k=0;
    for(int i=1;i<=n;i++)
        if(rk[p][i]==1)h[rk[p][i]]=0;
        else 
        {
            int j=sa[p][rk[p][i]-1];
            while(a[i+k]==a[j+k])k++;
            h[rk[p][i]]=k;if(k>0)k--;
        }
}
void solve()
{
    int ans=-1;
    for(int i=2;i<=n;i++) 
        if((sa[p][i]<=lena&&sa[p][i-1]>lena)||
          (sa[p][i]>lena&&sa[p][i-1]<=lena))
        ans=max(ans,h[i]);
    cout<<ans<<endl;
}
int main()
{
        scanf("%s%s",A+1,B+1);
        lena=strlen(A+1);lenb=strlen(B+1); n=lena+lenb;
        fo(i,1,lena) a[i]=A[i]-'a'+1;
        a[lena+1]=28;n++;
        fo(i,1,lenb) a[i+1+lena]=B[i]-'a'+1;
        getsa(); geth();
//      fo(i,1,n)printf("sa[p][%d]=%d\n",i,sa[p][i]);
        solve();
    return 0;
}