昨天试着写了一下后缀数组,蛋疼了……
今天又看了一下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;