【uoj #35】后缀排序

这是一道模板题

这是一道模板题。

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

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

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

#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的代码……