要求字符串中找出两个最大相同的公共子串
思路:
利用height数组,height数组最大的位置即为结果,另可通过sa[i]求取最大的位置。
#include<iostream>
#include<algorithm>
#include<stdio.h>
#include<string.h>
using namespace std;
/*
后缀数组模板,下标从0开始
s[]原字符串
后缀sa[i]和sa[j] LCP(i, j) = min{height[k] | min{i, j} < k <= max{i, j}}最长公共前缀
Heigth[i] : 表示Suffix[SA[i]]和Suffix[SA[i - 1]]的最长公共前缀,
也就是排名相邻的两个后缀的最长公共前缀
rank[i] 第i个后缀的排名; SA[i] 排名为i的后缀位置
*/
const int N=1000010;
char s[N];
int c[N],x[N],y[N],n,m=256,height[N],rk[N],sa[N];
void Suffix()
{
for(int i=0;i<m;i++) c[i]=0;
for(int i=0;i<n;i++) c[x[i]=s[i]]++;
for(int i=1;i<m;i++) c[i]+=c[i-1];
for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i;
for(int k=1;k<=n;k<<=1)
{
int p=0;
for(int i=n-k;i<n;i++) y[p++]=i;
for(int i=0;i<n;i++) if(sa[i]>=k) y[p++]=sa[i]-k;
for(int i=0;i<m;i++) c[i]=0;
for(int i=0;i<n;i++) c[x[y[i]]]++;
for(int i=1;i<m;i++) c[i]+=c[i-1];
for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i];
swap(x,y);
p=1;x[sa[0]]=0;
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;
}
}
void getheight()
{
for(int i=0;i<n;i++) rk[sa[i]]=i;
for(int i=0,k=0;i<n;i++)
{
if(rk[i])
{
if(k) --k;
//else k=0;
int j=sa[rk[i]-1];
while(s[i+k]==s[k+j]) k++;
height[rk[i]]=k;
}
}
}
int main()
{
scanf("%s",s);
n = strlen(s);
s[n++] = '0';
Suffix();
getheight();
int res = sa[0],len=1;
for(int i=1;i<n;i++)
{
if(height[i]>len){
len = height[i];
res = sa[i];
}
}
for(int i=res;i<=res+len-1;i++)
printf("%c",s[i]);
}