UOJ#35 后缀排序

这是一道模板题。

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

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

输入格式

一行一个长度为 n 的仅包含小写英文字母的字符串。

输出格式

第一行 n 个整数，第 i 个整数表示排名为 i 的后缀的第一个字符在原串中的位置。

第二行 n−1 个整数，第 i 个整数表示排名为 i 和排名为 i+1 的后缀的最长公共前缀的长度。

样例一

input

ababa

output

5 3 1 4 2

1 3 0 2

explanation

排序后结果为：

a
aba
ababa
ba
baba

限制与约定

1≤n≤105

时间限制：1s

空间限制：256MB

我的后缀平衡树终于过了

#include<cstdio>

#include<cctype>

#include<queue>

#include<ctime>

#include<cstring>

#include<algorithm>

#define mid l+r>>1

#define rep(s,t) for(int i=s;i<=t;i++)

#define ren for(int i=first[x];i!=-1;i=next[i])

using namespace std;

inline int read() {

    int x=,f=;char c=getchar();

    for(;!isdigit(c);c=getchar()) if(c=='-') f=-;

    for(;isdigit(c);c=getchar()) x=x*+c-'';

    return x*f;

}

const int maxn=;

typedef long long ll;

int h[maxn],ch[maxn][],c[maxn],tot,rt;

ll rk[maxn];

void rebuild(int o,ll l,ll r) {

    if(!o) return;rk[o]=l+r;

    rebuild(ch[o][],l,mid);rebuild(ch[o][],mid,r);

}

void rotate(int& o,int d,ll l,ll r) {

    int k=ch[o][d^];ch[o][d^]=ch[k][d];ch[k][d]=o;o=k;

    rebuild(o,l,r);

}

int cmp(int x,int y) {return c[x]<c[y]||(c[x]==c[y]&&rk[x-]<rk[y-]);}

void insert(int& o,ll l,ll r) {

    if(!o) {o=tot;rk[o]=l+r;return;}

    if(cmp(tot,o)) {insert(ch[o][],l,mid);if(h[ch[o][]]>h[o]) rotate(o,,l,r);}

    else {insert(ch[o][],mid,r);if(h[ch[o][]]>h[o]) rotate(o,,l,r);}

}

void insert(int x) {

    c[++tot]=x;h[tot]=rand()*rand();

    insert(rt,,1ll<<);

}

void print(int o) {

    if(!o) return;

    print(ch[o][]);

    printf("%lld ",rk[o]);

    print(ch[o][]);

}

int n,sa[maxn],rank[maxn],height[maxn];

int cmp2(int a,int b) {return rk[n-a+]<rk[n-b+];}

char s[maxn];

int main() {

    srand(time());

    scanf("%s",s);n=strlen(s);

    rep(,n) insert(s[n-i]-'a'+),sa[i]=i;

    sort(sa+,sa+n+,cmp2);

    rep(,n) rank[sa[i]]=i,printf("%d ",sa[i]);puts("");

    for(int i=,j,k=;i<=n;i++) {

        j=sa[rank[i]-];

        while(s[i+k-]==s[j+k-]) k++;

        height[rank[i]]=k?k--:k;

    }

    rep(,n) printf("%d ",height[i]);

    return ;

}

秒客网