后缀数组

博客:

https://www.cnblogs.com/zwfymqz/p/8413523.html

模板：

洛谷P3809

#include<bits/stdc++.h>
#define maxn 1000050
using namespace std;
char s[maxn]="0aabaaaab";
int y[maxn],x[maxn],c[maxn],sa[maxn],rk[maxn],height[maxn];
int n,m;	//sa[i] 第i小后缀的起始位置 rk[i] 后缀i的排名 height[i] 排名i和排名i-1的后缀的最长公共前缀

inline void get_SA()
{
    for(int i=1;i<=n;++i) ++c[x[i]=s[i]];
	//c数组是桶
	//x[i]是第i个元素的第一关键字
    for(int i=2;i<=m;++i) c[i]+=c[i-1]; //m 输入字符串的 ASCII 上限
	//做c的前缀和，我们就可以得出每个关键字最多是在第几名 计数排序的思想
    for(int i=n;i>=1;--i) sa[c[x[i]]--]=i;
    for(int k=1;k<=n;k<<=1)
	{
        int num=0;
        for(int i=n-k+1;i<=n;++i) y[++num]=i;
		//y[i]表示第二关键字排名为i的数，第一关键字的位置
		//第n-k+1到第n位是没有第二关键字的 所以排名在最前面 要先处理
        for(int i=1;i<=n;++i) if(sa[i]>k) y[++num]=sa[i]-k;
		//排名为i的数 在数组中是否在第k位以后
		//如果满足(sa[i]>k) 那么它可以作为别人的第二关键字，就把它的第一关键字的位置添加进y就行了
		//所以i枚举的是第二关键字的排名，第二关键字靠前的先入队
        for(int i=1;i<=m;++i) c[i]=0;
		//初始化c桶
        for(int i=1;i<=n;++i) ++c[x[i]];
		//因为上一次循环已经算出了这次的第一关键字 所以直接加就行了
        for(int i=2;i<=m;++i) c[i]+=c[i-1]; //第一关键字排名为1~i的数有多少个
        for(int i=n;i>=1;--i) sa[c[x[y[i]]]--]=y[i],y[i]=0;
		//因为y的顺序是按照第二关键字的顺序来排的
		//第二关键字靠后的，在同一个第一关键字桶中排名越靠后
		//基数排序
        swap(x,y);
		//这里不用想太多，因为要生成新的x时要用到旧的，就把旧的复制下来，没别的意思
        x[sa[1]]=1;
        num=1;
        for(int i=2;i<=n;++i)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) ? num:++num;
		//因为sa[i]已经排好序了，所以可以按排名枚举，生成下一次的第一关键字
        if(num==n) break;
        m=num;
		//这里就不用那个m了，因为都有新的编号了
    }
}
inline void get_height()
{
    int k=0;
    for(int i=1;i<=n;++i) rk[sa[i]]=i;
    for(int i=1;i<=n;++i)
	{
        if(rk[i]==1) continue;//第一名height为0
        if(k) --k;//h[i]>=h[i-1]-1;
        int j=sa[rk[i]-1];
        while(j+k<=n&&i+k<=n&&s[i+k]==s[j+k]) ++k;
        height[rk[i]]=k;//h[i]=height[rk[i]];
    }
}
int main()
{
	scanf("%s",s+1);
	n=strlen(s+1);
	m='z';
	get_SA();
	for(int i=1;i<=n;i++)
		printf("%d ",sa[i]);
	return 0;
}