【题目链接】 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: ",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 ",len[x]); else printf("%d ",min(ans,rmq_min(Rank[pos[x]],Rank[pos[y]]))); } }return 0; }