网上的算法看不懂,只有自己写个了,但只求出rank数组,sa数组和rank数组互逆可求出,不知道height数组怎么求。。。
代码有点乱。m=字符范围,如26个小写字母,n=字符串长度
复杂度为logn(m+kn)
#include<iostream>
using namespace std;
#include<string.h>
char str[9]="aabaaaab";
int SA[8];
struct rank_str
{
rank_str()
{}
rank_str(int x,int y)
{
x=x;
y=y;
}
int x;
int y;
int rank;
rank_str *next;
};
rank_str r[8];
rank_str t[8];
void cal_rank()
{
int i;
rank_str **head=new rank_str*[26];
rank_str **rear=new rank_str*[26];
for(i=0;i<26;i++)
{
head[i]=new rank_str;
head[i]->next=NULL;
rear[i]=head[i];
}
for(i=0;i<=7;i++)
{
rank_str *tmp=new rank_str;
tmp->x=t[i].x;
tmp->y=t[i].y;
tmp->next=NULL;
rear[t[i].y]->next=tmp;
rear[t[i].y]=tmp;
}
int c=0;
for(i=0;i<26;i++)
{
rank_str *p=head[i]->next;
while(p)
{
t[c].x=p->x;
t[c].y=p->y;
c++;
p=p->next;
}
}
head=new rank_str*[26];
rear=new rank_str*[26];
for(i=0;i<26;i++)
{
head[i]=new rank_str;
head[i]->next=NULL;
rear[i]=head[i];
}
for(i=0;i<=7;i++)
{
rank_str *tmp=new rank_str;
tmp->x=t[i].x;
tmp->y=t[i].y;
tmp->next=NULL;
rear[t[i].x]->next=tmp;
rear[t[i].x]=tmp;
}
c=0;
for(i=0;i<26;i++)
{
rank_str *p=head[i]->next;
while(p)
{
t[c].x=p->x;
t[c].y=p->y;
c++;
p=p->next;
}
}
t[0].rank=1;
for(i=1;i<=7;i++)
if(t[i].x==t[i-1].x&&t[i].y==t[i-1].y)
t[i].rank=t[i-1].rank;
else t[i].rank=t[i-1].rank+1;
}
void da(rank_str r[],int j)
{
int i;
for(i=0;i<=7-j;i++)
{
r[i].x=r[i].rank;
r[i].y=r[i+j].rank;
}
for(;i<=7;i++)
{
r[i].x=r[i].rank;
r[i].y=0;
}
}
int main()
{
int i;
for(i=0;i<8;i++)
{
r[i].rank=str[i]-'a'+1;
}
for(i=0;i<log(8)/log(2);i++)
{
da(r,pow(2,i));
for(int i=0;i<=7;i++)
{
t[i].x=r[i].x;
t[i].y=r[i].y;
}
cal_rank();
for(int k=0;k<=7;k++)
for(int j=0;j<=7;j++)
if(t[j].x==r[k].x&&t[j].y==r[k].y)
{
r[k].rank=t[j].rank;
break;
}
}
/*for(int k=0;k<=7;k++)
{
r[k].rank-=1;
cout<<r[k].rank;
}
cout<<endl;
for(int k=0;k<=7;k++)
SA[r[k].rank]=k;
for(int k=0;k<=7;k++)
cout<<SA[k];*/
}