牛客2020暑假第一场，A题后缀数组

https://ac.nowcoder.com/acm/contest/5666/A

A-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
思路：首先可以得到一个结论那就是对于这个字符串反过来之后得到的排序和应该得到的排序正好相反。也就是把之前的0变为n，然后再在字符串的末尾加一个n+1即可
从开头算距离下一个相等的字母的距离，如果没有则为n
原本的为abaab
序号 原后缀 对应的后缀Ci
1 abaab 231556
2 baab 31556
3 aab 1556
4 ab 556
5 b 56
得到的排序为3 1 2 4 5
接下来就是对于代码的实现了
得到了一个新的字符串，即可以用后缀数组直接求排名，输出sa即可
代码如下：

#include <cstdio>
#include <iostream>
using namespace std;
const int maxn=1e5+10;
int wa[maxn], wb[maxn], wv[maxn], ws1[maxn];
int cmp(int *r, int a, int b, int l)
{
    return r[a] == r[b] && r[a + l] == r[b + l];
}
void da(int *r, int *sa, int n, int m)
{
    int i, j, p, *x = wa, *y = wb, *t;
    for (i = 0; i < m; i++)
        ws1[i] = 0;
    for (i = 0; i < n; i++)
        ws1[x[i] = r[i]]++;
    for (i = 1; i < m; i++)
        ws1[i] += ws1[i - 1];
    for (i = n - 1; i >= 0; i--)
        sa[--ws1[x[i]]] = i;
    for (j = 1, p = 1; p < n; j *= 2, m = p)
    {
        for (p = 0, i = n - j; i < n; i++)
            y[p++] = i;
        for (i = 0; i < n; i++)
            if (sa[i] >= j)
                y[p++] = sa[i] - j;
        for (i = 0; i < n; i++)
            wv[i] = x[y[i]];
        for (i = 0; i < m; i++)
            ws1[i] = 0;
        for (i = 0; i < n; i++)
            ws1[wv[i]]++;
        for (i = 1; i < m; i++)
            ws1[i] += ws1[i - 1];
        for (i = n - 1; i >= 0; i--)
            sa[--ws1[wv[i]]] = y[i];
        for (t = x, x = y, y = t, p = 1, x[sa[0]] = 0, i = 1; i < n; i++)
            x[sa[i]] = cmp(y, sa[i - 1], sa[i], j) ? p - 1 : p++;
    }
    return;
}
char str[maxn];
int a[maxn], sa[maxn];
void init(int n){
     scanf("%s", str);
    int pa=-1, pb=-1;
    for(int i = n-1; i >= 0; i--){
        if(str[i]=='a'){
            if(pa!=-1) a[i] = pa-i;
            else a[i] = n;
            pa = i;
        }else {
            if(pb!=-1) a[i] = pb-i;
            else a[i] = n;
            pb = i;
        }
    }
    a[n] = n+1;
}
int main()
{
    int n;
     while(~scanf("%d", &n)){
        init(n);
        //因为之前在n上加了一个数，所以字符串长度加一
        n++;
        //板子，且最大值小于 m
        da(a,sa,n,n+1);
        for(int i = n-2; i >= 0; i--){ 
            printf("%d ", sa[i]+1);
        }
        printf("\n");
    } 
    return 0;
}