sa 板子 (sa、height)

最新推荐文章于 2024-06-13 10:13:20 发布

Snow_raw

最新推荐文章于 2024-06-13 10:13:20 发布

阅读量160

点赞数

分类专栏：算法模板大全文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_53504307/article/details/127308010

版权

算法模板大全专栏收录该内容

5 篇文章 0 订阅

订阅专栏

$s a$ 板子 $(s a 、 h e i g h t)$

$s a$ 定义： $s a [i]$ 记录排名为 $i$ 的非空后缀的编号。

$h e i g h t$ 定义： $h e i g h t [i]$ 来记录排名为 $i$ 的非空后缀与排名为 $i - 1$ 的非空后缀的最长公共前缀的长度 $(1 <= i <= n)$ ， $h e i g h t [1] = 0$ 。

宏定义：

const int N = 1000010;

int n, m;
char s[N];
int sa[N], x[N], y[N], c[N], rk[N], height[N];

获得 $s a :$

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

获得 $h e i g h t :$

void get_height()
{
    for (int i = 1; i <= n; i ++ ) rk[sa[i]] = i;
    for (int i = 1, k = 0; i <= n; i ++ )
    {
        if (rk[i] == 1) continue;
        if (k) k -- ;
        int j = sa[rk[i] - 1];
        while (i + k <= n && j + k <= n && s[i + k] == s[j + k]) k ++ ;
        height[rk[i]] = k;
    }
}

$main :$

int main()
{
    scanf("%s", s + 1);
    n = strlen(s + 1), m = 122;
    get_sa();
    get_height();

    for (int i = 1; i <= n; i ++ ) printf("%d ", sa[i]);
    puts("");
    for (int i = 1; i <= n; i ++ ) printf("%d ", height[i]);
    puts("");
    return 0;
}