标签:char bool 排序 tmp scan pre efi span cst
后缀数组板子,给出一个字符串,要求输出sa数组。
首先声明一下各数组的作用。
rank:序列上的点->点权;
sa:排名->位置;
tmp:辅助数组;
#include<cstdio> #include<cstring> #define N 1000050 int rank[N],tmp[N],sa[N],hs[N],num[N]; int n; char s[N]; bool cmp(int i,int j,int k) { if(i+k>n||j+k>n)return 0; return rank[i]==rank[j]&&rank[i+k]==rank[j+k]; } int main() { scanf("%s",s+1); n = strlen(s+1); int i,cnt=0; for(i=1;i<=n;i++)num[i]=s[i]; for(i=1;i<=n;i++)hs[num[i]]++; for(i=1;i<=200;i++)if(hs[i])tmp[i]=++cnt; for(i=1;i<=200;i++)hs[i]+=hs[i-1]; for(i=1;i<=n;i++)rank[i]=tmp[num[i]],sa[hs[num[i]]--]=i; for(int k=1;cnt!=n;k<<=1) { for(i=1;i<=n;i++)hs[i]=0; for(i=1;i<=n;i++)hs[rank[i]]++; for(i=1;i<=n;i++)hs[i]+=hs[i-1]; for(i=n;i>=1;i--)if(sa[i]>k)tmp[sa[i]-k]=hs[rank[sa[i]-k]]--; for(i=1;i<=k;i++)tmp[n-i+1]=hs[rank[n-i+1]]--; for(i=1;i<=n;i++)sa[tmp[i]]=i; for(i=1,cnt=0;i<=n;i++)tmp[sa[i]]=cmp(sa[i],sa[i-1],k)?cnt:++cnt; for(i=1;i<=n;i++)rank[i]=tmp[i]; } for(i=1;i<=n;i++)printf("%d ",sa[i]); printf("\n"); return 0; }
标签:char bool 排序 tmp scan pre efi span cst
原文地址:https://www.cnblogs.com/LiGuanlin1124/p/9670985.html