【uoj #35】后缀排序

来源:互联网 发布:淘宝好评语50字以上 编辑:程序博客网 时间:2024/05/16 03:39

这是一道模板题

这是一道模板题。

读入一个长度为 n 的由小写英文字母组成的字符串,请把这个字符串的所有非空后缀按字典序从小到大排序,然后按顺序输出后缀的第一个字符在原串中的位置。位置编号为 1n

除此之外为了进一步证明你确实有给后缀排序的超能力,请另外输出 n1 个整数分别表示排序后相邻后缀的最长公共前缀的长度。

直接上模板了……机房要关门了……我要吃饭了……

#include <iostream>#include <cstdio>#include <algorithm>#include <cstring>using namespace std;const int MAXN = 100000 + 5;int sa[MAXN],rank[MAXN],tmp[MAXN];int n,k;char c[MAXN];bool cmp_sa(int a,int b){    if(rank[a] != rank[b])        return rank[a] < rank[b];    else    {        int x = a + k <= n ? rank[a + k] : -1;        int y = b + k <= n ? rank[b + k] : -1;        return x < y;    }}void make_sa(char s[]){    for(int i = 0;i <= n;i ++)    {        rank[i] = i < n ? s[i] : -1;        sa[i] = i;    }    for(k = 1;k <= n;k <<= 1)    {        sort(sa,sa + n + 1,cmp_sa);        tmp[sa[0]] = 0;        for(int i = 1;i <= n;i ++)            tmp[sa[i]] = tmp[sa[i - 1]] + cmp_sa(sa[i - 1],sa[i]);        for(int i = 0;i <= n;i ++)            rank[i] = tmp[i];    }    return;}int lcp[MAXN];void make_lcp(char s[]){    for(int i = 1;i < n;i ++)        rank[sa[i]] = i;    int h = 0;    lcp[0] = 0;    for(int i = 0;i < n;i ++)    {        int j = sa[rank[i] - 1];        if(h)            h--;        for(;i + h < n && j + h < n;h ++)            if(s[i + h] != s[j + h])                break;        lcp[rank[i] - 1] = h;    }    return;}int main(){    scanf("%s",c);    n = strlen(c);    make_sa(c);    for(int i = 1;i <= n;i ++)        printf("%d ",sa[i] + 1);    puts("");    make_lcp(c);    for(int i = 1;i < n;i ++)        printf("%d ",lcp[i]);    return 0;}

//2016.4.7才发现这个是CE的代码……

0 0
原创粉丝点击