后缀数组

Posted 2020-10-10 star_eternal

namespace SA{
char s[N];
int sa[N],t1[N],t2[N],num[N];
int n;
void buildsa(int m)
{
int *x=t1,*y=t2;
for(int i=0;i<m;i++) num[i]=0;
for(int i=0;i<n;i++) num[x[i]=s[i]]++;
for(int i=1;i<m;i++) num[i]+=num[i-1];
for(int i=n-1;i>=0;i--) sa[--num[x[i]]]=i;
for(int k=1;k<=n;k<<=1)
{
int p=0;
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++) num[i]=0;
for(int i=0;i<n;i++) num[x[y[i]]]++;
for(int i=1;i<m;i++) num[i]+=num[i-1];
for(int i=n-1;i>=0;i--) sa[--num[x[y[i]]]]=y[i];
p=1;swap(x,y);
x[sa[0]]=0;
for(int i=1;i<n;i++)
x[sa[i]]=(y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k])?p-1:p++;
if(p>=n) break;
m=p;
}
}
int rank[N<<1],height[N<<1];
void getheight()
{
for(int i=0;i<n;i++) rank[sa[i]]=i;
int k=0;
for(int i=0;i<n;i++)
{
if(k) k--;
int j=sa[rank[i]-1];
while(s[j+k]==s[i+k]) k++;
height[rank[i]]=k;
}
}
}