LightOJ 1428 Melody Comparison

来源：互联网发布：陈奕迅蔡健雅知乎编辑：程序博客网时间：2024/05/04 05:28

刚开始提交的时候，惊讶地发现竟然超时了

检查了N遍终于发现把strlen写在了for循环里，这样算算时间复杂度就n^2了

解题思路：

把B串在A串出现的位置找出来

然后单独对A串计算一下height数组，然后枚举每个位置，求可以生成的子串，再减height就可以了

#include<stdio.h> #include<string.h>#include<algorithm>using namespace std;#define N 100005char s[N];int r[N],sa[N],height[N],rank[N],wa[N],wb[N],wv[N],ws[N]; int rmax[N];int cmp(int *r,int a,int b,int l){return r[a]==r[b]&&r[a+l]==r[b+l];}void da(int *r,int *sa,int n,int m){int i,j,p,*x=wa,*y=wb;for(i=0;i<m;i++) ws[i]=0;for(i=0;i<n;i++) ws[x[i]=r[i]]++;for(i=1;i<m;i++) ws[i]+=ws[i-1];for(i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;for(j=1,p=1;p<n;m=p,j<<=1){for(i=n-j,p=0;i<n;i++) y[p++]=i;for(i=0;i<n;i++) if (sa[i]>=j) y[p++]=sa[i]-j;for(i=0;i<n;i++) wv[i]=x[y[i]];for(i=0;i<m;i++) ws[i]=0;for(i=0;i<n;i++) ws[wv[i]]++;for(i=1;i<m;i++) ws[i]+=ws[i-1];for(i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];swap(x,y);for(p=1,x[sa[0]]=0,i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;}}void calheight(int *r,int *sa,int n){int i,j,k=0;for(i=1;i<=n;i++) rank[sa[i]]=i;for(i=0;i<n;height[rank[i++]]=k)for(k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);//for(i=1;i<=n;i++) printf("%d ",height[i]);}int main(){#ifndef ONLINE_JUDGEfreopen("in.txt","r",stdin);#endifint T;scanf("%d",&T);int cas=1;while(T--){int l=0;scanf("%s",s);int n1=strlen(s);for(int i=0;i<n1;i++) r[l++]=s[i];r[l++]=129;scanf("%s",s);int n2=strlen(s);for(int i=0;i<n2;i++) r[l++]=s[i];r[l]=0;da(r,sa,l+1,130);calheight(r,sa,l);memset(rmax,-1,sizeof(rmax));for(int i=rank[n1+1]+1;i<=l;i++) {if(height[i]>=n2) {rmax[sa[i]]=n2-1;}else break;}for(int i=l;i>=0;i--){if(rmax[i]==-1) rmax[i]=rmax[i+1]+1;//初值赋为-1是因为从-1开始刚刚好最后那几个数也符合这个公式 }r[n1]=0;da(r,sa,n1+1,130);calheight(r,sa,n1);//一开始做错了是因为没有重新计算height，放在后面的字符串影响了结果 //for(int i=0;i<n1;i++) printf("%d :%d      %d:%d  \n",i,rmax[i],i,height[rank[i]]);long long ans=0;for(int i=1;i<=n1;i++){//每次减去这个后缀的重复的子串数，因为height数组是和前一个的lcp，这样刚好还有一个没计算到的即为所求 int total=min(rmax[sa[i]],n1-sa[i]);if(total>height[i]) ans+=total-height[i];}printf("Case %d: %lld\n",cas++,ans);}return 0;}

0 0