后缀数组模板

学习后缀数组的模板

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>

using namespace std;
const int maxn = 2e6 + 10;
int sa[maxn],rk[maxn], height[maxn], h[maxn], y[maxn], x[maxn], c[maxn];
//sa【i】为排名为i的后缀首元素在字符串的位置，rk【i】为首元素在字符串中为i的后缀的排名
//height【i】为排名为i和排名为和排名为i - 1的lcp，h【i】为height的反数组
// c是第一关键字数量的前缀和
//y中间数组，循环前部分存储排名为i的第二关键字的位置，后部分存储上次的第一关键字
int n, m;
char s[maxn];
void get_sa()
{
    for(int i = 1; i <= n; i++) c[x[i] = s[i]]++;
    for(int i = 2; i <= m; i++)
        c[i] += c[i - 1];
    for(int i = n; i >= 1; i--) sa[c[x[i]]--] = i;
    //基数排序
    for(int k = 1; k <= n; k<<= 1)
    {
        int num = 0;
        for(int i = n - k + 1; i <= n; i++) y[++num] = i;
        for(int i = 1; i <= n; i++) if(sa[i] > k) y[++num] = sa[i] - k;
        for(int i = 0; i <= m; i++) c[i] = 0;
        for(int i = 1; i <= n; i++) c[x[i]]++;
        for(int i = 2; i <= m; i++) c[i] += c[i - 1];
        for(int i = n; i >= 1; i--) sa[c[x[y[i]]]--] = y[i], y[i] = 0;
        swap(x, y);
        x[sa[1]] = 1, num = 1;
        for(int i = 2; i <= n; i++)
        x[sa[i]] = (y[sa[i]] == y[sa[i - 1]] && y[sa[i] + k] == y[sa[i - 1] + k]) ? num : ++num;
        if(num == n) break;
        m = num;
    }
}
void get_height()
{
    int k = 0;
    for(int i = 1; i <= n; i++) rk[sa[i]] = i;
    for(int i = 1; i <= n; i++)
    {
        if(rk[i] == 1) continue;
        if(k) k--;//h【i】 >= h[i - 1] - 1
        int j = sa[rk[i] - 1];
        while(j + k <= n && i + k <= n && s[j + k] == s[i + k]) k++;
        height[rk[i]] = k;

    }
}

int main()
{
    scanf("%s", s + 1);
    m = 122, n = strlen(s + 1);
    get_sa();
    for(int i = 1; i <= n; i++)
    if(i == n)
    printf("%d\n", sa[i]);
    else
    printf("%d ", sa[i]);
    return 0;
}