后缀数组代码详解

//
//  main.cpp
//  后缀数组模板详解
//
//  Created by XD on 15/9/24.
//  Copyright © 2015年 XD. All rights reserved.
//

#include <iostream>
using namespace std ;
const int maxn = 10000 ;
int s[maxn] ;
int sa[maxn],t1[maxn] , t2[maxn] ,c[maxn] ;
void getSa(int n,int m )
{
int i , j ;
int *x = t1 , *y = t2 ,*temp;
for(int i = 0 ; i <=m ; i++) c[i] =0 ;
for(int i = 0 ; i < n ; i++) c[x[i] = s[i]]++ ;
for(i = 1 ; i <=m ; i++) c[i] += c[i-1] ;
for(int i = n-1 ; i >-1 ; i -- ) sa[--c[x[i]]] = i ;
//下面是最关键的一段代码
for(int k = 1 ; k <= n ; k*= 2 )
{
int p  =0  ;
//先计数排序第二关键字，这里不需要重复计数排序的步骤，因为在上一次的排序过后，是第一关键字的顺序，可是根据倍增算法的特点，这一次的第二关键字的顺序与上一次的排序的第一关键字（注意有的位置后面是没有第二关键字的）

//这是因为从 n -  k到n-1的位置上第二关键字都是没有的。这是设为最小的，因为当一个字符串是另一个字符串的前缀时，后面的子符串是较大的。这里还有一个问题，有人说可能第一关键字都是不完整的，那为什么可以比较大小呢。因为我们在原字符串的末尾加了一个$，我们认为这个$符号是最小的，又用因为后缀的长度是不同的，所以在仅有的长度下就可以比较出大小，因此即使有的关键字没达到k的长度，但是任然是可行的。
for(i = n - k ;  i < n ; i ++ ) y[p++] = i ;
//这里是对具有第二关键字的那部分排序，当然不能忽略不具有第一关键字的那部分，因此才有sa[i]>=k.
for(i = 0 ; i < n ; i ++) if(sa[i]>=k) y[p++] = sa[i] - k ;

//对第一关键字排序
for(i = 0 ; i <=m; i++) c[i] = 0 ;
//这里是x[y[i]]的原因是对第二关键字排序后，需按照其结果的对第一关键字排序，而y[i]存的是第一关键字的位置
for(i = 0 ; i < n ;i++) c[x[y[i]]]++ ;
for(i = 1 ;i<=m;i++) c[i] += c[i-1] ;
for(i = n - 1; i > -1;i--) sa[--c[x[y[i]]]] = y[i] ;
//下面要更新x数组
temp = x ; x = y ; y = temp ;
x[sa[0]] = 1 ; p=1;
for(i = 1 ; i < n ;i++)
{
//根据排序确定下一次的x数组的值
x[sa[i]] = (y[sa[i]] == y[sa[i-1]] && y[sa[i] + k]==y[sa[i-1] + k ])? p : ++p  ;
}
if(p >= n){
break ;
}
m = p ;
}
}
int main(int argc, const char * argv[]) {
int n ;
while (scanf("%d" , &n),n) {
for (int i= 0 ; i < n ; i++) {
scanf("%d" , &s[i]) ;
s[i]++ ;
}
s
=1 ;
getSa(n+1 , 100) ;
for (int i = 0; i <= n; i++) {
printf("%d ",sa[i]) ;
}
}
return 0;
}