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

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;
}