【uoj 35】后缀排序

最新推荐文章于 2024-01-02 16:24:10 发布

WeZiH

最新推荐文章于 2024-01-02 16:24:10 发布

阅读量425

点赞数

分类专栏：后缀数组

本文链接：https://blog.csdn.net/Qantun_Mechanics/article/details/52958704

版权

后缀数组专栏收录该内容

2 篇文章 0 订阅

订阅专栏

没什么好说的确实是后缀数组的模板题目。
感觉后缀数组理解起来还是不容易啊，嗯，看了有段时间吧。今天还算是过了模板题。

代码细节多，而且不好理解的（如果不自己手动模拟的话），其实背代码也未尝不可。嘿嘿嘿。

代码：

#include <iostream>
#include <cstring>
#include <cstdio>
#include <string>

const int MAXN = 100005;

using namespace std;

char s[MAXN];
int sa[MAXN],r[MAXN],t[MAXN],t2[MAXN],c[MAXN],h[MAXN],n;

inline void build_sa(int m)
{
    int *x = t,*y = t2;

    for(int i = 0;i < m;i++) c[i] = 0;
    for(int i = 0;i < n;i++) c[x[i] = s[i]]++;
    for(int i = 0;i < m;i++) c[i] += c[i-1];
    for(int i = n - 1;i >= 0;i--) sa[--c[x[i]]] = i;

    for(int k = 1;k <= n;k <<= 1){
        int p = 0;
        for(int i = n - k;i < n;i++) y[p++] = i;
        for(int i = 0;i < n;i++) if(sa[i] >= k) y[p++] = sa[i] - k;

        for(int i = 0;i < m;i++) c[i] = 0;
        for(int i = 0;i < n;i++) c[x[y[i]]]++;
        for(int i = 0;i < m;i++) c[i] += c[i-1];
        for(int i = n - 1;i >= 0;i--) sa[--c[x[y[i]]]] = y[i];

        swap(x,y);
        p = 1;x[sa[0]] = 0;
        for(int i = 1;i < n;i++)
            x[sa[i]] = y[sa[i-1]] == y[sa[i]] && y[sa[i-1]+k] == y[sa[i]+k] ? p - 1 : p++;
        if(p >= n) break;
        m = p;  
    } 

}

inline void build_Height()
{
    int i,j,k = 0;
    for(i = 0;i < n;i++) r[sa[i]] = i;
    for(i = 0;i < n;i++)
    {
        if(k) k--;
        j = sa[r[i] - 1];
        while(s[i+k] == s[j + k]) ++k;
        h[r[i]] = k;
    }
}

int main()
{
    scanf("%s",s);

    n = strlen(s) + 1;

    build_sa(256);

    for(int i = 1;i < n;i++)
    printf("%d ",sa[i]+1);

    build_Height();

    printf("\n");
    for(int i = 2;i < n;i++)
    printf("%d ",h[i]);

    return 0;
}