尾碼數組模版

代碼來自 演算法競賽入門經典。

#include <cmath>#include <ctime>#include <iostream>#include <string>#include <vector>#include <cstdio>#include <cstdlib>#include <cstring>#include <queue>#include <map>#include <set>#include <algorithm>#include <cctype>#include <stack>#include <deque>using namespace std;typedef long long LL;#define EPS 10e-9#define INF 0x3f3f3f3f#define REP(i,n) for(int i=0; i<(n); i++)const int maxn = 10000;char s[maxn];int sa[maxn],t[maxn],t2[maxn],c[maxn],n;void build_sa(int m){     int i,*x=t,*y=t2;     //基數排序     for(int i=0;i<m;i++)  c[i]=0;     for(int i=0;i<n;i++)  c[x[i]=s[i]-'a' ]++;     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;           //直接利用sa數組排序第二關鍵字           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];           //根據sa和y數組計算新的x數組           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);    build_sa(26);    for(int i=0;i<n;i++) printf("%d ",sa[i]);    return 0;}