POJ 1226 Substrings(后缀数组+二分答案)
【题目链接】 http://poj.org/problem?id=1226
【题目大意】
求在每个给出字符串中出现的最长子串的长度,字符串在出现的时候可以是倒置的。
【题解】
我们将每个字符串倒置,用拼接符和原串拼接,然后将所有通过这种方式得到的字符串拼接,做后缀数组,因为求最长子串,所以我们考虑二分答案后检验,将单串的拼接串视为一个串,我们在检验时只要判断是否在每个串中出现过即可。
【代码】
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 | # include <cstdio> # include <cstring> # include <vector> # include <algorithm> using namespace std; const int N= 200010 ; int n,m,rank[N],sa[N],h[N],tmp[N],cnt[N],ans,a[N],s[N]; char str[N]; void suffixarray( int n, int m){ int i,j,k;n++; for (i= 0 ;i< 2 *n+ 5 ;i++)rank[i]=sa[i]=h[i]=tmp[i]= 0 ; for (i= 0 ;i<m;i++)cnt[i]= 0 ; for (i= 0 ;i<n;i++)cnt[rank[i]=s[i]]++; for (i= 1 ;i<m;i++)cnt[i]+=cnt[i- 1 ]; for (i= 0 ;i<n;i++)sa[--cnt[rank[i]]]=i; for (k= 1 ;k<=n;k<<= 1 ){ for (i= 0 ;i<n;i++){ j=sa[i]-k; if (j< 0 )j+=n; tmp[cnt[rank[j]]++]=j; }sa[tmp[cnt[ 0 ]= 0 ]]=j= 0 ; for (i= 1 ;i<n;i++){ if (rank[tmp[i]]!=rank[tmp[i- 1 ]]||rank[tmp[i]+k]!=rank[tmp[i- 1 ]+k])cnt[++j]=i; sa[tmp[i]]=j; }memcpy(rank,sa,n*sizeof( int )); memcpy(sa,tmp,n*sizeof( int )); if (j>=n- 1 ) break ; } for (j=rank[h[i=k= 0 ]= 0 ];i<n- 1 ;i++,k++) while (~k&&s[i]!=s[sa[j- 1 ]+k])h[j]=k--,j=rank[sa[j]+ 1 ]; } int len[N],u,K; vector< int > S[N]; bool vis[N]; bool check( int L){ int cur=- 1 ; for ( int i= 1 ;i<=u;i++){ if (h[i]<L)S[++cur].clear(); S[cur].push_back(i); } for ( int i= 0 ;i<=cur;i++){ if (S[i].size()>=n){ memset(vis, 0 ,sizeof(vis)); for ( int j= 0 ;j<S[i].size();j++){ int k=S[i][j]; int x=upper_bound(a,a+n+ 1 ,sa[k])-a- 1 ; vis[x]= 1 ; } int count= 0 ; for ( int j= 0 ;j<n;j++) if (vis[j])count++; if (count>=n) return 1 ; } } return 0 ; } int T; int main(){ scanf( "%d" ,&T); while (T--){ scanf( "%d" ,&n); int tmp= 200 ; u= 0 ; for ( int i= 0 ;i<n;i++){ scanf( "%s" ,str); len[i]=strlen(str); for ( int j= 0 ;j<len[i];j++)s[u++]=( int )str[j]; s[u++]=tmp++; for ( int j=len[i]- 1 ;j>= 0 ;j--)s[u++]=( int )str[j]; s[u++]=tmp++; }tmp= 0 ; s[u]= 0 ; for ( int i= 0 ;i<=n;i++){ a[i]=tmp; if (i<n)tmp=tmp+(i== 0 ? 2 *len[i]+ 1 : 2 *len[i]+ 2 ); }suffixarray(u, 410 ); int l= 1 ,r= 100 ,ans= 0 ; while (l<=r){ int mid=(l+r)>> 1 ; if (check(mid))ans=mid,l=mid+ 1 ; else r=mid- 1 ; } if (n== 1 )printf( "%d\n" ,len[ 0 ]); else printf( "%d\n" ,ans); } return 0 ; } |