【uoj 35】后缀排序

来源:互联网 发布:在香港用淘宝 编辑:程序博客网 时间:2024/06/05 02:23

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

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

代码:

#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;} 
0 0
原创粉丝点击