一道用后缀数组求LCP的模板题,后缀数组用的还是不够熟,细节掌握的还是不够好,RE了好多次……
#include <iostream>
#include <cstdio>
#include <string>
#include <cctype>
#include <algorithm>
using namespace std;
#define M 10010
string str;
int s[M],sa[M],t[M],t2[M],c[M],n;
int rank[M],height[M];
void build_sa(int m)
{
int i,*x=t,*y=t2;
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(int 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=0;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 i,j,k=0;
for(i=0;i<n;i++) rank[sa[i]]=i;
for(i=0;i<n;i++)
{
if(k) k--;
int j=sa[rank[i]-1];
while(s[i+k]==s[j+k]) k++;
height[rank[i]]=k;
}
}
int main()
{
while(getline(cin,str))
{
if(str=="") break;
n=0;
for(int i=0;i<str.size();i++)
{
if(!isupper(str[i])) continue;
s[n++]=str[i]-'A'+1;
}
s[n++]=0;
build_sa(27);
getheight();
//for(int i=0;i<n;i++) cout<<sa[i]<<" "<<height[i]<<endl;
int l=1,cnt;
do
{
cnt=0;
int tmp=1;
for(int i=1;i<n;i++)
{
if(height[i]>=l) tmp++;
else
{
cnt=max(tmp,cnt);
tmp=1;
}
}
cnt=max(cnt,tmp);
if(cnt>1) cout<<cnt<<endl;
l++;
}
while(cnt>1);
cout<<endl;
}
return 0;
}