后缀数组模板 SuffixArray

后缀数组模板 SuffixArray

#include<bits/stdc++.h>
#define rep(i,a,b) for(i=a;i<b;++i)
#define drep(i,a,b) for(i=b;i>=a;--i)
using namespace std;
int p[N],wc[N],wv[N],wa[N],wb[N],rank[N],sa[N];
char s[N];
inline bool cmp(int *r,int a,int b,int l) {
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
inline void tate(int *x,int *y,char *r,int n,int m,bool opt) {
    int i;
    rep(i,0,m) wc[i]=0;
    rep(i,0,n) wc[opt?(x[i]=r[i]):wv[i]]++;
    rep(i,1,m) wc[i]+=wc[i-1];
    drep(i,n-1,0) sa[--wc[opt?x[i]:wv[i]]]=opt?i:y[i];
}
void f(char *r,int *sa,int n,int m) {
    int i,j,p,*x=wa,*y=wb,*t;
    tate(x,y,r,n,m,1);
    for (j=1,p=1;p<n;j<<=1,m=p) {
        p=0;
        rep(i,n-j,n) y[p++]=i;
        rep(i,0,n) if(sa[i]>=j) y[p++]=sa[i]-j;
        rep(i,0,n) wv[i]=x[y[i]];
        tate(x,y,r,n,m,0);
        for (t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;++i)
            x[sa[i]]=cmp(r,sa[i-1],sa[i],j)?p-1:p++;
    }
}

压了下，把基数排序的压进函数，然后看起来还挺好写的，这只是后缀数组的一个阶段啦， 继续努力！