UVA 12338:Anti-Rhyme Pairs(後綴數組+ST表)
阿新 • • 發佈:2017-07-15
後綴數組 min -1 class break nbsp con mem span
【題目鏈接】 click
【題目大意】
給出一些字符串,詢問查詢任意兩個字符串的最長公共前綴
【題解】
將字符串拼接,對拼接的字符串做後綴數組,對於查詢的兩個字符串,
只要在height數組上查詢區間最小值即可。
特別註意多組數據時候對字符串結尾的處理,很久沒寫容易忽視導致wa。
【代碼】
#include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int N=4000010; int n,m,Rank[N],sa[N],h[N],tmp[N],cnt[N],ans;char t[N],s[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 f[N][30],lg2[N]; void rmq_init(int n){ for(int i=2;i<=n;i++)lg2[i]=lg2[i/2]+1; for(int i=1;i<=n;i++)f[i][0]=h[i]; for(int j=1;(1<<j)<=n;j++) for(int i=1;i+(1<<j)-1<=n;i++) f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]); } int rmq_min(int l,int r){ if(l>r)swap(l,r);l++; int k=lg2[r-l+1]; return min(f[l][k],f[r-(1<<k)+1][k]); } int T,pos[N],len[N]; int main(){ scanf("%d",&T); int cas=1; while(T--){ printf("Case %d:\n",cas++); scanf("%d",&n); int u=1; for(int i=1;i<=n;i++){ scanf("%s",&t); len[i]=strlen(t); pos[i]=u; for(int j=0;j<len[i];j++)s[u++]=t[j]; }s[u]=0; //特別註意多組數據s[u]=0!!!! suffixarray(u,256); rmq_init(u); int q; scanf("%d",&q); while(q--){ int x,y; scanf("%d%d",&x,&y); int ans=min(len[x],len[y]); if(x==y)printf("%d\n",len[x]); else printf("%d\n",min(ans,rmq_min(Rank[pos[x]],Rank[pos[y]]))); } }return 0; }
UVA 12338:Anti-Rhyme Pairs(後綴數組+ST表)