emmmmm后缀数组还没系统开始学习,所以借这道题先大体了解一下后缀数组,以后会慢慢补完的。
所以在这里思路参考的这篇博客,算是我能大体看懂得了,太菜了…
代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 1e5+7;
//先当做一个简易的后缀数组模板 后期再添加 height和lcp
char s[MAXN];
int b[MAXN],pos[MAXN],ans[MAXN],flag[MAXN],len[MAXN];
int nxt[MAXN][2];
int cnt[MAXN],rak[MAXN<<1],tmp[MAXN<<1],sa[MAXN];
struct Suffix_Array
{
// tmp数组是基数排序时的辅助数组,cnt代表的就是桶
int r,n;//n代表字符串的长度 r代表字符串中的最大值加1
void radix_sort(int *rank,int *tp){
memset(cnt,0,sizeof(cnt[0])*(r+10));
for(int i = 1;i <= n;i ++) cnt[rank[tp[i]]]++;
for(int i = 1;i <= r;i ++) cnt[i] += cnt[i-1];
for(int i = n;i >= 1;i --) sa[cnt[rank[tp[i]]]--] = tp[i];
}
void suffix(){
int *rk = rak,*tp = tmp;
for(int i = 1;i <= n;i ++){
rk[i] = b[i]+1,tp[i] = i,tp[i+n] = rk[i+n] = 0;
}
radix_sort(rk,tp);
for(int p = 0,k = 1, i;p < n;k<<=1,r = p){
for(p = 0,i = n-k+1;i <= n;i ++) tp[++p] = i;
for(int i = 1;i <= n;i ++) if(sa[i] > k) tp[++p] = sa[i] - k;
radix_sort(rk,tp);
swap(rk,tp);
rk[sa[1]] = 1;
p = 1;
for(int i = 2;i <= n;i ++){
if(tp[sa[i-1]] != tp[sa[i]] || tp[sa[i]+k] != tp[sa[i-1]+k])
p++;
rk[sa[i]] = p;
}
}
for(int i = 1;i <= n;i ++) rak[sa[i]] = i;
}
}SA;
bool cmp(int x,int y){
int lenx = len[x],leny = len[y];
if(lenx != leny) return lenx < leny;//第一个a和第一个b组成的01序列不相等直接返回长度 就是字典序的大小
if(flag[x] || flag[y]) return flag[x] < flag[y];
return rak[x+lenx+1] < rak[y+leny+1];
}
int main()
{
int n;
while(~scanf("%d",&n)){
scanf("%s",s+1);
int t = 0;
pos[0] = pos[1] = 0;
for(int i = 1;i <= n;i ++){
ans[i] = i;
int x = s[i]-'a';
b[i] = 0;
if(pos[x]) b[i] = i - pos[x];
pos[x] = i;
t = max(t,b[i]);//找字符串中最大的值
}
nxt[n+1][0] = nxt[n+1][1] = n+1;
for(int i = n;i >= 1;i --){
int x = s[i]-'a';
nxt[i][x] = i;
nxt[i][x ^ 1] = nxt[i+1][x ^ 1];
len[i] = -(nxt[i][x]-nxt[i][x ^ 1]);
if(nxt[i][x ^ 1] == n+1){
flag[i] = 1;len[i]--;
}
}
SA.n = n;
SA.r = t+1;
SA.suffix();
sort(ans+1,ans+1+n,cmp);
for(int i = 1;i <= n;i ++){
flag[i] = 0;
printf(i==n?"%d\n":"%d ",ans[i]);
}
}
return 0;
}