SA模板(DA||DC3)

DA版本的SA速度相较DC3慢了些，很容易被卡，但是他在所需空间上优越于DC3

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize(4)
#include<bits/stdc++.h>
using namespace std;
using namespace std;
typedef long long ll;
typedef long long LL;
#define rank Rank
const int maxn=1e6+5;
int n,pos;
int tot;
int wa[maxn],wb[maxn],wsf[maxn],wv[maxn],sa[maxn],r[maxn];
int rank[maxn],height[maxn];
char s[maxn];
bool cmp(int *r,int a,int b,int k)
{
    return r[a]==r[b]&&r[a+k]==r[b+k];
}
void get_sa_da(int *r,int *sa,int n,int m)
{
    int i,j,p,*x=wa,*y=wb,*t;
    for(i=0; i<m; i++) wsf[i]=0;
    for(i=0; i<n; i++) wsf[x[i]=r[i]]++;
    for(i=1; i<m; i++) wsf[i]+=wsf[i-1];
    for(i=n-1; i>=0; i--) sa[--wsf[x[i]]]=i;
    p=1,j=1;
    for(; 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++) wsf[i]=0;
        for(i=0; i<n; i++) wsf[wv[i]]++;
        for(i=1; i<m; i++) wsf[i]+=wsf[i-1];
        for(i=n-1; i>=0; i--) sa[--wsf[wv[i]]]=y[i];
        t=x,x=y,y=t;
        x[sa[0]]=0;
        for(p=1,i=1; i<n; i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)? p-1:p++;
    }
}
void get_height(int *r,int n)
{
    int i,j,k=0;
    for(i=0; i<n; i++) rank[sa[i]]=i;
    for(i=0; i<n; i++)
    {
        if(k) k--;
        else k=0;
        j=sa[rank[i]-1];
        while(r[i+k]==r[j+k]) k++;
        height[rank[i]]=k;
    }
}
int main()
{
    while(~scanf("%s",s)&&s[0]!='.')
    {
        int Max=0,ans=1;
        n=strlen(s);
        for(int i=0;i<n;i++) r[i]=s[i],Max=max(Max,r[i]);
        r[n]=0;
        get_sa_da(r,sa,n+1,200);
        get_height(r,n+1);
        for(int i=1;i<=n;i++)
        {
            if(n%i) continue;
            if(rank[0]-rank[i]!=1) continue;
            if(height[rank[0]]!=n-i) continue;
            ans=n/i;
            break;
        }
        printf("%d\n",ans);
    }
}

 

DC3虽然时间性能较好，但是遇到RMQ型的问题，可能被卡空间

#pragma GCC optimize(2)
#pragma GCC optimize(3)
#pragma GCC optimize(4)
#include<bits/stdc++.h>
#define rank Rank
#define F(x) ((x)/3+((x)%3==1?0:tb))
#define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)
const int maxn=1e6+5;
int n;
char s[maxn];
int sa[maxn],rank[maxn],height[maxn];
int r[maxn],wa[maxn],wb[maxn],wv[maxn],wws[maxn];
void sort(int *r,int *a,int *b,int n,int m)
{
    int i;
    for(i=0; i<n; i++) wv[i]=r[a[i]];
    for(i=0; i<m; i++) wws[i]=0;
    for(i=0; i<n; i++) wws[wv[i]]++;
    for(i=1; i<m; i++) wws[i]+=wws[i-1];
    for(i=n-1; i>=0; i--) b[--wws[wv[i]]]=a[i];
    return;
}
int c0(int *r,int a,int b)
{
    return r[a]==r[b]&&r[a+1]==r[b+1]&&r[a+2]==r[b+2];
}
int c12(int k,int *r,int a,int b)
{
    if(k==2) return r[a]<r[b]||r[a]==r[b]&&c12(1,r,a+1,b+1);
    else return r[a]<r[b]||r[a]==r[b]&&wv[a+1]<wv[b+1];
}
void get_sa_dc3(int *r,int *sa,int n,int m)
{
    int i,j,*rn=r+n,*san=sa+n,ta=0,tb=(n+1)/3,tbc=0,p;
    r[n]=r[n+1]=0;
    for(i=0; i<n; i++) if(i%3!=0) wa[tbc++]=i;
    sort(r+2,wa,wb,tbc,m);
    sort(r+1,wb,wa,tbc,m);
    sort(r,wa,wb,tbc,m);
    for(p=1,rn[F(wb[0])]=0,i=1; i<tbc; i++) rn[F(wb[i])]=c0(r,wb[i-1],wb[i])?p-1:p++;
    if(p<tbc) get_sa_dc3(rn,san,tbc,p);
    else for(i=0; i<tbc; i++) san[rn[i]]=i;
    for(i=0; i<tbc; i++) if(san[i]<tb) wb[ta++]=san[i]*3;
    if(n%3==1) wb[ta++]=n-1;
    sort(r,wb,wa,ta,m);
    for(i=0; i<tbc; i++) wv[wb[i]=G(san[i])]=i;
    for(i=0,j=0,p=0; i<ta && j<tbc; p++) sa[p]=c12(wb[j]%3,r,wa[i],wb[j])?wa[i++]:wb[j++];
    for(; i<ta; p++) sa[p]=wa[i++];
    for(; j<tbc; p++) sa[p]=wb[j++];
    return;
}
void get_height(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 main()
{
    while(~scanf("%s",s)&&s[0]!='.')
    {
        int Max=0,ans=1;
        n=strlen(s);
        for(int i=0;i<n;i++) r[i]=s[i],Max=max(Max,r[i]);
        r[n]=0;
        get_sa_dc3(r,sa,n+1,Max+1);
        get_height(r,sa,n);
        for(int i=1;i<=n;i++)
        {
            if(n%i) continue;
            if(rank[0]-rank[i]!=1) continue;
            if(height[rank[0]]!=n-i) continue;
            ans=n/i;
            break;
        }
        printf("%d\n",ans);
    }
    return 0;
}