版权声明:欢迎转载,转载请注明出处,如有错误,还望指出,谢谢 博客地址:https://blog.csdn.net/lanyanzhiji123asd https://blog.csdn.net/lanyanzhiji123asd/article/details/89522648
后缀数组扩展
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<set>
#include<queue>
using namespace std;
#define ll long long
int n,m,kk;
const int ding=2e4+5;
const int INF=0x3f3f3f3f;
int sa[ding],x[ding],y[ding],c[ding],r[ding],rankk[ding],height[ding];
void get_SA()
{
int i,k;
// memset(x,0,sizeof(x));//因为我是以a来作为终止符,所以后面的空 应该设置成-1
// memset(y,0,sizeof(y));
memset(c,0,sizeof(c));//清空桶;
for(i=0;i<n;i++) c[x[i]=r[i]]++;
for(i=1;i<m;i++) c[i]+=c[i-1];
for(i=n-1;i>=0;i--)
{
sa[--c[x[i]]]=i;
}
int p=1;
for(k=1;k<n;k<<=1,m=p)
{
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;//大于等于k
//sa是第一关键字的排名, 然后再根据第二关键字的排名来确定排名,
memset(c,0,sizeof(c));
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]; //重置sa数组
memcpy(y,x,sizeof(x));
memset(x,0,sizeof(x));
// swap(x,y);
p=1;//从1开始编号
x[sa[0]]=0;
//因为上面已经重置了x的值 所以下面是从1开始循环,然后是和前一个比较sa[i]和sa[i-1] 比较
for(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++;
//x数组里面记住的是x[i]=j 下标为i的后缀,排名为j
//y数组的内容是y[i]=j 下标为j 的后缀排名为i
if(p>=n)break;
}
// for(i=0;i<n;i++)
// cout<<sa[i]<<" ";
// cout<<endl;
}
void get_height()
{
int i,k=0,j;
for(i=0;i<n;i++) rankk[sa[i]]=i;
for(i=0;i<n-1;i++)
{//为什么循环到n-1,因为我们在原始字符串后面加了a,所以这个后缀一定是最小的。
if(k)k--;
j=sa[rankk[i]-1];
for(;r[i+k]==r[j+k];k++);
height[rankk[i]]=k;
}
// for(i=0;i<n-1;i++)
// cout<<height[i]<<" ";
// cout<<endl;
}
int main()
{
std::ios::sync_with_stdio(false);
string str;
int len,ans,i,loc,maxn,minn;
cin>>str;
len=str.length();
for(i=0;i<str.length();i++)
r[i]=(int)(str[i]);
r[len]=299;//中间设置一个较大的值,不知道为什么
n=len+1;
for(i=len-1;i>=0;i--)
r[n++]=r[i];
r[n]=0;
n++;
// for(i=0;i<n;i++)
// cout<<r[i]<<" ";
// cout<<endl;
m=ding;
get_SA();
get_height();
ans=1;
loc=0;
for(i=2;i<n;i++)
{
if(height[i]>=ans)
{
maxn=max(sa[i],sa[i-1]);
minn=min(sa[i],sa[i-1]);
if(maxn<=len||minn>=len)continue;
if(len-(minn+height[i]-1)!=maxn-len)continue;//abcdecba 这个样例注意下
if(height[i]==ans)
{
loc=min(minn,loc);
continue;
}
ans=height[i];
loc=minn;
}
}
for(i=loc;i<loc+ans;i++)
cout<<str[i];
cout<<endl;
return 0;
}