洛谷P3809 【模板】后缀排序

最新推荐文章于 2024-05-11 22:52:39 发布

weixin_30460489

最新推荐文章于 2024-05-11 22:52:39 发布

阅读量94

点赞数

原文链接：http://www.cnblogs.com/whisperlzw/p/11420384.html

版权

本文详细介绍了一种高效的字符串处理算法——后缀排序(SA)算法，并提供了完整的C++实现代码。该算法能够快速地对字符串的所有后缀进行排序，适用于解决字符串模式匹配等问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【模板】后缀排序

题目传送门

代码如下

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;

inline int read(){
    int res = 0, w = 0; char ch = 0;
    while(!isdigit(ch)){
        w |= ch == '-', ch = getchar();
    }
    while(isdigit(ch)){
        res = (res << 3) + (res << 1) + (ch ^ 48);
        ch = getchar();
    }
    return w ? -res : res;
}

const int N = 1000005;

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

void get_sa()
{
    for(int i = 1; i <= m; i ++) c[i] = 0;
    for(int i = 1; i <= n; i ++) c[x[i] = s[i]] ++;
    for(int i = 1; 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 = 1; i <= m; i ++) c[i] = 0;
        for(int i = 1; i <= n; i ++) c[x[i]] ++;
        for(int i = 1; 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);
        num = 1;
        x[sa[1]] = num;
        for(int i = 2; i <= n; i ++)
            if(sa[i] + k <= n && sa[i - 1] + k <= n)
                x[sa[i]] = (y[sa[i]] == y[sa[i - 1]] && y[sa[i] + k] == y[sa[i - 1] + k])? num: ++num;
            else
                x[sa[i]] = ++num;
        if(num == n)
            break;
        m = num;
    }
    for(int i = 1; i <= n; i ++){
        if(i != 1)
            putchar(' ');
        printf("%d", sa[i]);
    }
    putchar('\n');
}

int main()
{
    gets(s + 1);
    n = strlen(s + 1);
    m = 'z';
    get_sa();
    return 0;
}

转载于:https://www.cnblogs.com/whisperlzw/p/11420384.html