【模板】后缀数组&后缀自动机

关于后缀自动机SAM，贴一个非常好的讲解：后缀自动机学习小记
代码交洛谷模板可A

---

后缀数组

传送门：洛谷-【模板】后缀排序

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
char s[N];
int sa[N],t1[N],t2[N],c[N];
int n,m=555;

void suffix()
{
    int *x=t1,*y=t2;
    for(int i=0;i<m;i++) c[i]=0;
    for(int i=0;i<n;i++) c[x[i]=s[i]]++;
    for(int i=1;i<m;i++) c[i]+=c[i-1];
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
    for(int k=1;k<=n;k<<=1){
        int p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=0;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(int i=1;i<n;i++) 
          x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
         if(p>=n) break;
         m=p; 
    }
}

int main(){
    scanf("%s",s);
    n=strlen(s);
    suffix();
    for(int i=0;i<n;i++) printf("%d ",sa[i]+1);
    return 0; 
}

---

后缀自动机

传送门：【模板】后缀自动机

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+10;
const int M=2e6+20;
char S[N];ll ans;
int cnt=1,cur=1,last,len;
int dis[M],sa[M],c[N];

struct bk{
    int fa,ch[26],size;
}t[M];

inline void insert(int a,int d)
{
    last=cur;cur=++cnt;dis[cur]=d;
    int p=last;
    for(;p && !t[p].ch[a];p=t[p].fa) t[p].ch[a]=cur;
    if(!p) t[cur].fa=1;else{
        int q=t[p].ch[a];
        if(dis[q]==dis[p]+1) t[cur].fa=q;else{
            int nt=++cnt;dis[nt]=dis[p]+1;
            memcpy(t[nt].ch,t[q].ch,sizeof(t[q].ch));
            t[nt].fa=t[q].fa;t[q].fa=nt;t[cur].fa=nt;
            for(;t[p].ch[a]==q;p=t[p].fa) t[p].ch[a]=nt;
        }
    }
    t[cur].size=1;
}

void query()
{
    for(int i=1;i<=cnt;i++) c[dis[i]]++;
    for(int i=1;i<=len;i++) c[i]+=c[i-1];
    for(int i=cnt;i>=1;i--) sa[c[dis[i]]--]=i;
    for(int i=cnt;i>=1;i--){
        int p=sa[i];
        if(t[p].size>1) ans=max(ans,(ll)t[p].size*dis[p]);
        t[t[p].fa].size+=t[p].size;
    }
}

int main(){
   scanf("%s",S+1);
   len=strlen(S+1);
   for(int i=1;i<=len;i++) insert(S[i]-'a',i);
   query();
   printf("%lld\n",ans); 
   return 0;
}