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

做法参考：博客

Code：

#include<bits/stdc++.h>
const int maxn=1e4 + 10;
using namespace std;
int rk[maxn],wb[maxn],sa[maxn];
int tmp[maxn],h[maxn],height[maxn];

int a[maxn];
int n, vis[3];
char t[maxn];

void getsa()
{
	
	int *x =rk,*y=tmp;
	int s = n + 1;
	for (int i=1;i<=n;i++) x[i]=a[i],y[i]=i;
	for (int i=1;i<=s;i++) wb[i]=0;
	for (int i=1;i<=n;i++) wb[x[y[i]]]++;
	for (int i=1;i<=s;i++) wb[i]+=wb[i-1];
	for (int i=n;i>=1;i--) sa[wb[x[y[i]]]--]=y[i];
	int p =0;
	for (int j=1;p<n;j<<=1){
		p=0;
		for (int i=n-j+1;i<=n;i++) y[++p]=i;
		for (int i=1;i<=n;i++) if (sa[i]>j) y[++p] = sa[i]-j;
		for (int i=1;i<=s;i++) wb[i]=0;
		for (int i=1;i<=n;i++) wb[x[y[i]]]++;
		for (int i=1;i<=s;i++) wb[i]+=wb[i-1];
    	for (int i=n;i>=1;i--) sa[wb[x[y[i]]]--]=y[i];
		swap(x,y);
		p=1;
		x[sa[1]]=1;
		for (int i=2;i<=n;i++){
			x[sa[i]] = (y[sa[i]]==y[sa[i-1]] && y[sa[i]+j]==y[sa[i-1]+j]) ? p : ++p; 
		} 
		s=p;
 	}
	for (int i=1;i<=n;i++) rk[sa[i]]=i;
	h[0]=0;
    for (int i=1;i<=n;i++){
		h[i]=max(h[i-1]-1,0);
		while(i+h[i]<=n && sa[rk[i]-1]+h[i]<=n && a[i+h[i]]==a[sa[rk[i]-1]+h[i]]) h[i]++;
	} 
	for (int i=1;i<=n;i++) height[i]=h[sa[i]];
}
 
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']) a[i] = n;
            else a[i] = vis[t[i]-'a'] - i;
            
            vis[t[i]-'a'] = i;
        }
        a[n + 1] = n + 1;
        ++n;
        getsa();
        for(int i = n - 1; i >= 1; --i) printf("%d ", sa[i]);
        puts("");
    }
}