2020牛客多校第一场A题B-Suffix Array（数据卡在1%的原因）

B-Suffix Array

题意

给定一个字符串长度和字符串，已知B函数定义为

说人话就是从字符串第一个字符开始，出现的第一个a或b对应的下标的b数组为0，否则就是离前面最近的和自己相同的字符的距离。
如

5
abaab

它的b数组就是00213
这就是说每个字符串都有对应的b数组

现在我们要求你求出给定字符串的所有后缀字符串的b数组，并对它们进行字典序从小到大排序。
我们称后缀i是从i开始的字符串后缀
输出字典序排序第一的是后缀几，排序第二的是后缀几……

如abaab
他的所有后缀字符串和对应的b数组为
abaab
00213 排第4
baab
0013排第3
aab
010排第5
ab
00排第2
b
0 排第1

所以答案为5 4 2 1 3

思路

这题有很多做法，我实在想不出其他方法，只能用官方结论来做
就是构造一个c数组
方法是

我是在这篇博客上看懂的
大佬博客

大概就是在c数组最后添一位n+1，n是字符串长度
然后从后往前遍历，出现的第一个a或b对应的下标的b数组为n，否则就是离后面最近的和自己相同的字符的距离
如abaab
他的所有后缀字符串和对应的c数组为
abaab
2 3 1 5 5 6 排第2
baab
3 1 5 5 6排第3
aab
1 5 5 6排第1
ab
5 5 6排第4
b
5 6 排第5
还有一个多出来的
6排第6
所以答案为3 1 2 4 5 6
反过来再不考虑6
得5 4 2 1 3
就是答案

这是结论，我不会证明

所以思路就是把字符串的c数组求出来，然后求它的后缀数组即可

关于后缀数组可以看
后缀数组

特别注意

需要特别注意的是这个后缀数组的初始数组不能是char数组，因为字符串元素会大于char的范围！！！！
也就是我代码里的s数组用int的原因！！！！
不然会卡在1%的数据。
我找这个错找了两天

代码

#include<iostream>
#include<cstring>
using namespace std;
const int N=1e5+10;
int n;
char cao[N];
int s[N];
int sa[N],t[N],t2[N],c[N];
void build_sa(int m){
	int i,*x=t,*y=t2;
	for(i=0;i<m;i++)c[i]=0;
	for(i=0;i<n;i++)c[x[i]=s[i]]++;
	for(i=0;i<m;i++)c[i]+=c[i-1];
	for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
	for(int k=1;k<=n;k<<=1){
		int p=0;
		
		for(i=n-k;i<n;i++)y[p++]=i;
		for(i=0;i<n;i++)if(sa[i]>=k)y[p++]=sa[i]-k;
		
		for(i=0;i<m;i++)c[i]=0;
		for(i=0;i<n;i++)c[x[y[i]]]++;
		for(i=0;i<m;i++)c[i]+=c[i-1];
		for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
		
		swap(x,y);
		p=1,x[sa[0]]=0;
		for(i=1;i<n;i++)
			x[sa[i]]=y[sa[i-1]]==y[sa[i]]&&y[sa[i-1]+k]==y[sa[i]+k]?p-1:p++;
		if(p>=n)break;
		m=p;
	}
}

int main(){
	while(cin>>n>>cao){
		int posa=-1,posb=-1;
		s[n]=n+1;
		for(int i=n-1;i>=0;i--){
			if(cao[i]=='a'){
				if(posa==-1){
					s[i]=n;
					posa=i;
				}else{
					s[i]=posa-i;
					posa=i;
				}
			}else {
				if(posb==-1){
					s[i]=n;
					posb=i;
				}else{
					s[i]=posb-i;
					posb=i;
				}
			}
		}
		++n;
		build_sa(n+1);
		for(int i=n-2;i>=0;i--){
			cout<<sa[i]+1<<" ";
		}
		cout<<endl;
	}
	return 0;
} 
/*
4
aaaa
4
baaa

5
abaab
*/