【POJ2774】Long Long Message，第一次的后缀数组

Time:2016.05.21

Author:xiaoyimi

转载注明出处谢谢

传送门

思路：没什么值得说的，大家想看思路的话还是去看大牛们的吧，磨了将近一天的后缀数组，基本按照模板打得，最后发现有些bug，又去博采众长才艰难的A掉，WA了不知道多少次啊啊啊

注意：

使用指针交换数组元素是很快的 ——by Yveh

代码：

#include<cstdio>
#include<iostream>
#define M 200004
using namespace std;
int cnt[M],w[M],sa[M],rank[M],height[M],tmp[M];
void SA(int len,int up)
{
int *rk=rank,p=0,*id=height,*t=tmp,d=1;
for (int i=0;i<up;i++) cnt[i]=0;
for (int i=0;i<len;i++) cnt[rk[i]=w[i]]++;
for (int i=1;i<up;i++) cnt[i]+=cnt[i-1];
for (int i=len-1;i>=0;i--) sa[--cnt[rk[i]]]=i;
for (;;)
{
for (int i=len-d;i<len;i++) id[p++]=i;
for (int i=0;i<len;i++)
if (sa[i]>=d) id[p++]=sa[i]-d;
for (int i=0;i<up;i++) cnt[i]=0;
for (int i=0;i<len;i++) cnt[t[i]=rk[id[i]]]++;
for (int i=0;i<up;i++) cnt[i]+=cnt[i-1];
for (int i=len-1;i>=0;i--) sa[--cnt[t[i]]]=id[i];
swap(t,rk);
p=1;
rk[sa[0]]=0;
for (int i=0;i<len-1;i++)
if (sa[i]+d<len&&sa[i+1]+d<len&&t[sa[i]]==t[sa[i+1]]&&t[sa[i]+d]==t[sa[i+1]+d])
rk[sa[i+1]]=p-1;
else rk[sa[i+1]]=p++;
if (p>=len) break;
d<<=1;up=p;p=0;
}
}
void Height(int len)
{
int i,j,k=0;
for(i=1; i<=len; i++)  rank[sa[i]]=i;
for(i=0; i<len; i++)
{
if(k)
k--;
else
k=0;
j=sa[rank[i]-1];
while(w[i+k]==w[j+k])
k++;
height[rank[i]]=k;
}
}
main()
{
int len=0,up=0;
char ch=getchar();
while (ch>'z'||ch<'a') ch=getchar();
while (ch>='a'&&ch<='z') w[len++]=ch-'a'+1,ch=getchar();
while (ch>'z'||ch<'a') ch=getchar();
int pos=len;
w[len++]=28;
while (ch>='a'&&ch<='z') w[len++]=ch-'a'+1,ch=getchar();
w[len]=0;
SA(len+1,30);
Height(len);
int ans=0;
for (int i=1;i<len;i++)
if ((sa[i]<pos)^(sa[i-1]<pos))
ans=max(height[i],ans);
printf("%d\n",ans);
}