poj3450 Corporate Identity(后缀数组+二分答案)

来源:互联网 发布:淘宝金牌卖家入口 编辑:程序博客网 时间:2024/05/20 17:28
题意:

求多个串的最长公共子串

把所有串都接在一起。。用特殊字符隔开。然后和两个串的一样,二分答案,每次判断这个长度是否可行。(分组,一组中出现在所有串中即满足

我以前写的SA太弱了。。还TLE了。。改进了一下下。

#include <cstdio>#include <cstring>#define N 805000int mm,m,n,ansl=0,a[N],id[N];int rank[N<<1],rank1[N],tmp[N],count[N],sa[N],h[N];inline int min(int x,int y){return x<y?x:y;}inline bool jud(int x){int cnt=0;bool vis[4005];for(int i=1;i<=n;++i){if(h[i]>=x){if(!vis[id[sa[i]]]) cnt++,vis[id[sa[i]]]=1;}else{memset(vis,0,sizeof(vis));vis[id[sa[i]]]=1;cnt=1;}if(cnt==mm){ansl=sa[i-1];return 1;}}return 0;}int main(){//freopen("a.in","r",stdin);while(1){scanf("%d",&mm);if(!mm) break;char s[210];int l=1,r=300;n=0,m=30;for(int i=1;i<=mm;++i){scanf("%s",s+1);int len=strlen(s+1);r=min(r,len);for(int j=1;j<=len;++j) a[++n]=s[j]-'a'+1,id[n]=i;a[++n]=++m;id[n]=i;}memset(rank,0,sizeof(rank1));for(int i=1;i<=m;++i) count[i]=0;for(int i=1;i<=n;++i) count[a[i]]=1;for(int i=1;i<=m;++i) count[i]+=count[i-1];for(int i=n;i>=1;--i) rank[i]=count[a[i]];int k=0;for(int p=1;k!=n;p<<=1,m=k){for(int i=1;i<=m;++i) count[i]=0;for(int i=1;i<=n;++i) count[rank[i+p]]++;for(int i=1;i<=m;++i) count[i]+=count[i-1];for(int i=n;i>=1;--i) tmp[count[rank[i+p]]--]=i;for(int i=1;i<=m;++i) count[i]=0;for(int i=1;i<=n;++i) count[rank[tmp[i]]]++;for(int i=1;i<=m;++i) count[i]+=count[i-1];for(int i=n;i>=1;--i) sa[count[rank[tmp[i]]]--]=tmp[i];memcpy(rank1,rank,sizeof(rank1));rank[sa[1]]=k=1;for(int i=2;i<=n;++i){if(rank1[sa[i]]!=rank1[sa[i-1]]||rank1[sa[i]+p]!=rank1[sa[i-1]+p]) ++k;rank[sa[i]]=k;}}k=0;for(int i=1;i<=n;++i){if(rank[i]==1){h[1]=0;continue;}if(i==1||h[rank[i-1]]<=1) k=0;if(k) --k;while(a[i+k]==a[sa[rank[i]-1]+k]) ++k;h[rank[i]]=k;}int ans=0;ansl=0;while(l<=r){int mid=(l+r)>>1;if(jud(mid)) ans=mid,l=mid+1;else r=mid-1;}if(l==1) puts("IDENTITY LOST");else{for(int i=0;i<ans;++i) putchar(a[ansl+i]+'a'-1);puts("");} }return 0;}


原创粉丝点击