1. 程式人生 > >UVA 12338:Anti-Rhyme Pairs(後綴數組+ST表)

UVA 12338:Anti-Rhyme Pairs(後綴數組+ST表)

後綴數組 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表)