cogs249 最长公共子串(后缀数组 二分答案

http://cogs.pro:8080/cogs/problem/problem.php?pid=pxXNxQVqP

题意:给m个单词,让求最长公共子串的长度。

思路:先把所有单词合并成一个串(假设长度是n,包含分隔符),中间用不同符号分隔,求出high[i](表示rk为i的和rk为i+1的后缀的最长公共前缀),然后二分答案ans,对于rk从1扫到n,如果有一段连续的rk值使得high[rk]>=ans且这段的串盖满了每个单词块,那么ans成立,即最终答案大于ans。

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const int N=20005;
int s[N],n=0,k,tmp[N],sa[N],rk[N],high[N],color[N],m,l[6];char S[6][2005];
bool comp(int i,int j){
if(rk[i]!=rk[j])return rk[i]<rk[j];
int ri=i+k<=n?rk[i+k]:-1;
int rj=j+k<=n?rk[j+k]:-1;
return ri<rj;
}
void Getsa(){
for(int i=1;i<=n;i++){
sa[i]=i;rk[i]=s[i];
}
for(k=1;k<=n;k<<=1){
sort(sa+1,sa+n+1,comp);
for(int i=1;i<=n;i++)tmp[sa[i]]=tmp[sa[i-1]]+comp(sa[i-1],sa[i]);
for(int i=1;i<=n;i++)rk[i]=tmp[i];
}
}
void Gethight(){
int j,h=0;
for(int i=1;i<=n;i++){
j=sa[rk[i]-1];
if(h)h--;
for(;j+h<=n && i+h<=n;h++)if(s[i+h]!=s[j+h])break;
high[rk[i]-1]=h;
}
}
void prework(int m){
int p=0;
for(int i=1;i<=m;i++){
for(int j=1;j<=l[i];j++)
p++,color[p]=i;
p++;
}
}
bool d[8];
bool check(int lim){
int p,ret;
for(int i=1;i<n;i++){
if(high[i]>=lim){
p=i;
while(p<n && high[p+1]>=lim)p++;p++;
for(int j=1;j<=m;j++)d[j]=false;
for(int j=i;j<=p;j++)d[color[sa[j]]]=true;
ret=p;p=1;
while(p<=m && d[p])p++;
if(p==m+1)return true;
i=ret;
}
}
return false;
}
int main()
{
freopen("pow.in","r",stdin);
freopen("pow.out","w",stdout);
int r=2e8;
scanf("%d",&m);
for(int i=1;i<=m;i++){
scanf("%s",S[i]);
l[i]=strlen(S[i]);
if(l[i]<r)r=l[i];
for(int j=0;j<l[i];j++)s[++n]=S[i][j];
s[++n]=i;
}
Getsa();Gethight();prework(m);
int l=0,mid,ans;
while(l<=r){
mid=(l+r)>>1;
if(check(mid))ans=mid,l=mid+1;
else r=mid-1;
}
printf("%d\n",ans);
return 0;
}
上一篇:java 实现DFA 算法(理论百度搜索)


下一篇:深入理解DOM事件机制系列第一篇——事件流