POJ1226 Substrings(二分+后缀数组)

题意:给n个字符串,求最长的子串,满足它或它的逆置出现在所有的n个字符串中。

  • 把n个字符串及其它们的逆置拼接,中间用不同字符隔开,并记录suffix(i)是属于哪个字符串的;
  • 跑后缀数组计算height;
  • 二分答案,height分组,看组里面是否都包含了n个字符串的后缀;
  • 注意n=1的情况。。
 #include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAXN 22222
int wa[MAXN],wb[MAXN],wv[MAXN],ws[MAXN];
int cmp(int *r,int a,int b,int l){
return r[a]==r[b] && r[a+l]==r[b+l];
}
int sa[MAXN],rank[MAXN],height[MAXN];
void SA(int *r,int n,int m){
int *x=wa,*y=wb; for(int i=; i<m; ++i) ws[i]=;
for(int i=; i<n; ++i) ++ws[x[i]=r[i]];
for(int i=; i<m; ++i) ws[i]+=ws[i-];
for(int i=n-; i>=; --i) sa[--ws[x[i]]]=i; int p=;
for(int j=; p<n; j<<=,m=p){
p=;
for(int i=n-j; i<n; ++i) y[p++]=i;
for(int i=; i<n; ++i) if(sa[i]>=j) y[p++]=sa[i]-j;
for(int i=; i<n; ++i) wv[i]=x[y[i]];
for(int i=; i<m; ++i) ws[i]=;
for(int i=; i<n; ++i) ++ws[wv[i]];
for(int i=; i<m; ++i) ws[i]+=ws[i-];
for(int i=n-; i>=; --i) sa[--ws[wv[i]]]=y[i];
swap(x,y); x[sa[]]=; p=;
for(int i=; i<n; ++i) x[sa[i]]=cmp(y,sa[i-],sa[i],j)?p-:p++;
} for(int i=; i<n; ++i) rank[sa[i]]=i;
int k=;
for(int i=; i<n-; height[rank[i++]]=k){
if(k) --k;
for(int j=sa[rank[i]-]; r[i+k]==r[j+k]; ++k);
}
} int sn,n,r[MAXN],belong[MAXN];
bool isok(int len){
bool vis[]={}; int cnt=;
for(int i=; i<=n; ++i){
if(height[i]>=len){
if(!vis[belong[sa[i]]]){
vis[belong[sa[i]]]=;
++cnt;
}
if(!vis[belong[sa[i-]]]){
vis[belong[sa[i-]]]=;
++cnt;
}
if(cnt==sn) return ;
}else{
if(cnt==sn) return ;
memset(vis,,sizeof(vis));
cnt=;
}
}
return ;
}
int main(){
char str[];
int t;
scanf("%d",&t);
while(t--){
n=;
scanf("%d",&sn);
for(int i=; i<sn; ++i){
scanf("%s",str);
if(sn==){
printf("%d\n",strlen(str));
break;
}
for(int j=; str[j]; ++j){
belong[n]=i;
r[n++]=str[j];
}
r[n++]=+(i<<);
for(int j=strlen(str)-; j>=; --j){
belong[n]=i;
r[n++]=str[j];
}
r[n++]=+(i<<|);
}
if(sn==) continue;
r[--n]=;
SA(r,n+,);
int l=,r=;
while(l<r){
int mid=l+r+>>;
if(isok(mid)) l=mid;
else r=mid-;
}
printf("%d\n",l);
}
return ;
}
上一篇:POJ1743 Musical Theme(二分+后缀数组)


下一篇:HDU4080Stammering Aliens(后缀数组+二分)