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[n]必定为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[N],t2[N],t1[N],sa[N],height[N],c[N],rank[N];
char str[N];
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[n]=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;
}