[luogu] P3809 【模板】后缀排序 (SA)
板子,照着题解打的倍增版。
#include <iostream> #include <cstdio> #include <cstring> using namespace std;const int N=1e6+7; char s[N];int n,m,x[N],y[N],bot[N],sa[N]; int main() { gets(s+1);n=strlen(s+1);m=122; for(int i=1;i<=n;i++) bot[x[i]=s[i]]++; for(int i=1;i<=m;i++) bot[i]+=bot[i-1]; for(int i=n;i>=1;i--) sa[bot[x[i]]--]=i; for(int k=1;k<=n;k<<=1) { int num=0; for(int i=n-k+1;i<=n;i++) y[++num]=i; for(int i=1;i<=n;i++) if(sa[i]>k) y[++num]=sa[i]-k; for(int i=1;i<=m;i++) bot[i]=0; for(int i=1;i<=n;i++) bot[x[i]]++; for(int i=1;i<=m;i++) bot[i]+=bot[i-1]; for(int i=n;i>=1;i--) sa[bot[x[y[i]]]--]=y[i],y[i]=0; swap(x,y);x[sa[1]]=1;num=1; for(int i=2;i<=n;i++) x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]) ? num : ++num; if(num==n) break;m=num; } for(int i=1;i<=n;i++) cout<<sa[i]<<" "; return 0; }