后缀数组。。。

一个简单的倍增法的后缀数组的实现。。。。 

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

const int maxn=1000;

char s[maxn];
int sa[maxn],t[maxn],t2[maxn],c[maxn],n;

void build_sa()
{
    int *x=t,*y=t2;
    ///基数排序1
    int m=26,n=strlen(s);
    for(int i=0;i<m;i++) c[i]=0;//每个字符初始
    for(int i=0;i<n;i++) c[x[i]=s[i]-'a']++;//每种字符出现的次数
    for(int i=1;i<m;i++) c[i]+=c[i-1];//小于等于字符i的数量
    for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;//排序

    for(int k=1;k<=n;k<<=1)//倍增
    {
        ///利用sa对第2关键字排序
        int p=0;
        for(int i=n-k;i<n;i++) y[p++]=i;//第二关键字都是0
        for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;//大于k的第2关键字

	///基数排序2
        for(int i=0;i<m;i++) c[i]=0;
        for(int i=0;i<n;i++) c[x[y[i]]]++;
        for(int i=0;i<m;i++) c[i]+=c[i-1];
        for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];

	///计算新的x数组
        swap(x,y);
        p=1; x[sa[0]]=0;//p不同的名次的个数
        for(int i=1;i<n;i++)
            x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+k]==y[sa[i-1]+k]?p-1:p++;
        if(p>=n) break;
        m=p;
    }
}

int main()
{
    cin>>s;
    build_sa();
    for(int i=0;i<strlen(s);i++) cout<<" "<<sa[i];
    return 0;
}


你可能感兴趣的:(Algorithm,后缀数组)