tyvj 1860 后缀数组入门题

描述 Description

我们定义一个字符串的后缀suffix(i)表示从s[i]到s[length(s)]这段子串。

后缀数组（Suffix array）SA[i]中存放着一个排列，满足suffix(sa[i])<suffix(sa[i+1]) 按照字典序方式比较

定义height[i]表示suffix(sa[i])与suffix(sa[i-1])之间的最长公共前缀长度，其中height[1]=0

你的任务就是求出SA和height这两个数组。字符串长度<=200000

输入格式 InputFormat

一行，为描述中的字符串（仅会出现小写字母）

输出格式 OutputFormat

共两行，每行n个数，第一行为sa[i]，第二行为height[i]，其中每行的数均用空格隔开
aabaaaab

4 5 6 1 7 2 8 3

0 3 2 3 1 2 0 1
http://www.tyvj.cn/problem_show.aspx?id=1860

#include<stdio.h>
#include<iostream>
#include<string.h>
#define N 200005
/*
*suffix array
*倍增算法  O(n*logn)
*待排序数组长度为n,放在0~n-1中，在最后面补一个0
*build_sa( ,n+1, );//注意是n+1;
*getHeight(,n);
*例如：
*n = 8;
*num[]   = { 1, 1, 2, 1, 1, 1, 1, 2, $ };注意num最后一位为0，其他大于0
*rank[]  = { 4, 6, 8, 1, 2, 3, 5, 7, 0 };rank[0~n-1]为有效值，rank
必定为0无效值
*sa[]    = { 8, 3, 4, 5, 0, 6, 1, 7, 2 };sa[1~n]为有效值，sa[0]必定为n是无效值
*height[]= { 0, 0, 3, 2, 3, 1, 2, 0, 1 };height[2~n]为有效值
*
*/
using namespace std;
int s
,t2
,t1
,sa
,height
,c
,rank
;
char str
;
void build_sa(int *s,int n,int m)
{
int *x=t1,*y=t2,i,k;
for(i=0;i<m;i++) c[i]=0;
for(i=0;i<n;i++) c[x[i]=s[i]]++;
for(i=1;i<m;i++) c[i]+=c[i-1];
for(i=n-1;i>=0;i--)  sa[--c[x[i]]]=i;
for(k=1;k<=n;k<<=1)
{
int p=0;
for(i=n-k;i<n;i++)  y[p++]=i;
for(i=0;i<n;i++)  if(sa[i]>=k)  y[p++]=sa[i]-k;
for(i=0;i<m;i++) c[i]=0;
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>=0;i--) sa[--c[x[y[i]]]]=y[i];
swap(x,y);
p=1;  x[sa[0]]=0;
for(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;
}
}
void getheight(int n)
{
int k=0,i,j;
for(i=0;i<=n;i++)
rank[sa[i]]=i;
for(i=0;i<n;i++)
{
if(k) k--;
j=sa[rank[i]-1];
while(s[i+k]==s[j+k]) k++;
height[rank[i]]=k;
}
}
int main()
{
int i,n;
while(scanf("%s",str)!=EOF)
{
n=strlen(str);
for(i=0;str[i]!='\0';i++)
s[i]=str[i]-'a'+1;
s
=0;
build_sa(s,n+1,30);
getheight(n);
for(i=1;i<=n;i++)
{
if(i==1)
printf("%d",sa[i]+1);
else
printf(" %d",sa[i]+1);
}
printf("\n");
for(i=1;i<=n;i++)
{
if(i==1)
printf("%d",height[i]);
else
printf(" %d",height[i]);
}
printf("\n");
}
return 0;
}