2020牛客暑期多校训练营（第一场）(A-B-Suffix Array 后缀数组 )

B-Suffix Array

做法参考来自：博客

题意：给你一个整数n，一个字符串B长度为n，其中对于n个字符串B的后缀设为Bi，但是这个后缀有点不一样，Bi=min1≤j<i,sj=si{i−j}，如果不存在这样的 j，那么 Bi=0，先求Bi按照字典序排序，从小到大输出排名
举个例子就比如 字符串为abaab
那么这个字符串拥有的后缀为
序号 原后缀 对应的后缀Bi
1 abaab 00213
2 baab 0013
3 aab 010
4 ab 00
5 b 0
如果对对应的后的Bi排序那就是 5 4 2 1 3

这个结论其实猜的出来，只是没能想到，也就没办法去证明。转化某种符合 后缀数组的形式，那就假设倒过求后缀bi，得到的排名是相反的，然后举例子证明下就可以了。

#include<bits/stdc++.h>
const int N=1e6 + 10;
using namespace std;
int s[N];
int y[N],x[N],c[N],sa[N],rk[N],height[N];
int n, vis[3];
char t[N];

inline void get_SA() {
    for(int i=0;i<=n+10;++i){
        c[i]=x[i]=y[i]=sa[i]=rk[i]=height[i]=0;
    }
    int m = n + 1; //ascll('z')=122

	for (int i=1; i<=n; ++i) ++c[x[i]=s[i]];//桶
	for (int i=2; i<=m; ++i) c[i]+=c[i-1];//求前缀和
	for (int i=n; i>=1; --i) sa[c[x[i]]--]=i;//逆着标序 得到sa
	for (int k=1; k<=n; k<<=1){ //倍增合并
		int num=0;
		for (int i=n-k+1; i<=n; ++i) y[++num]=i;//没有第二关键字，排名在最前
		for (int i=1; i<=n; ++i) if (sa[i]>k) y[++num]=sa[i]-k;

		for (int i=1; i<=m; ++i) c[i]=0;       //初始化桶
		for (int i=1; i<=n; ++i) ++c[x[i]];    //桶
		for (int i=2; i<=m; ++i) c[i]+=c[i-1]; //求前缀和
		for (int i=n; i>=1; --i) sa[c[x[y[i]]]--]=y[i],y[i]=0;//第二关键字真正的排了序

		for(int i=1;i<=n;++i) y[i] = x[i];
		
		x[sa[1]] = num = 1;
		for (int i=2; i<=n; ++i) //对第二关键字进行了简单的桶分类保存至x数组中
			x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) ? num : ++num;

		if (num==n) break;
		m=num;
	}
}

int main() {
    while(~scanf("%d", &n)){

        scanf("%s", t + 1);
        for(int i = 0; i <= 1; ++i) vis[i] = 0;
        for(int i = n; i >= 1; --i){
            if(!vis[t[i]-'a']) s[i] = n;
            else s[i] = vis[t[i]-'a'] - i;
            vis[t[i]-'a'] = i;
        }
        s[n + 1] = n + 1;
        ++n;
        get_SA();
        for(int i = n - 1; i >= 1; --i) printf("%d ", sa[i]);
        puts("");
    }
}