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

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

今天又看了一下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;