【省选】【后缀数组】模板

昨天试着写了一下后缀数组，蛋疼了……

今天又看了一下LHQ大神的pdf，感触很深

又把代码看了一遍，又理解了一下

把昨天的代码修改了一下，把两个for改成了memset，去掉了一个所谓的wv数组（他用wv数组只是后面可以直接用wv[i]表示x[y[i]]），直接改成嵌套成x[y[i]]

代码模板

【原数组 s 要在最后加上一个 ' $ ' 】

int sa[N],rank[N];//Suffix_Array数组、rank数组int c[N],tmp[N];//基数排序数组、辅助数组int height[N];//著名的height数组bool cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}void DA(int n,int m)//n为长度，m为字符串的最大值（基数排序用）{int i,j,p,*x=rank,*y=tmp;memset(c,0,sizeof(c));for(i=0;i<n;i++) c[x[i]=s[i]]++;for(i=1;i<m;i++) c[i]+=c[i-1];for(i=n-1;i>=0;i--) sa[--c[x[i]]]=i;for(j=1,p=0;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;memset(c,0,sizeof(c));for(i=0;i<n;i++) c[x[y[i]]]++;for(i=1;i<m;i++) c[i]+=c[i-1];for(i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];//不理解为什么是x[y[i]]，不过昨天达成了x[i]好像也过了……for(swap(x,y),p=1,x[sa[0]]=0,i=1;i<n;i++)if(!x[i]) continue;else x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;//这里有点不太理解}int k=0;//开始处理height数组for(i=0;i<n;height[x[i++]]=k)for(k?k--:0,j=sa[x[i]-1];s[j+k]==s[i+k];k++);//这里还是有点不太理解}

最后的rank数组应该为 *x ，所以如果要在当前函数之外使用rank数组的话，需要把rank处理一下

for(int i=0;i<n;i++) rank[sa[i]]=i;