【UOJ 35】后缀排序

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

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

代码:

#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;
} 
Last modification:October 1st, 2018 at 08:13 am
If you think my article is useful to you, please feel free to appreciate

Leave a Comment