3693 求這串字元中的重複次數最多的連續重複子串,多組答案輸出字典序最小的那個串(字尾陣列)
阿新 • • 發佈:2019-02-17
題目:求這串字元中的重複次數最多的連續重複子串,多組答案輸出字典序最小的那個串。
思路:與前一個題目幾乎一樣的,加上了字典序。多判斷就好
//#include<bits/stdc++.h> #include<iostream> #include<cstdio> #include<cstring> #include<algorithm> using namespace std; const int maxn=100010; int t1[maxn],t2[maxn],c[maxn]; bool cmp(int *r,int a,int b,int l) { return r[a]==r[b]&&r[a+l]==r[b+l]; } void da(int str[],int sa[],int ra[],int height[],int n,int m) { n++; int p,*x=t1,*y=t2; for(int i=0;i<m;i++) c[i]=0; for(int i=0;i<n;i++) c[x[i]=str[i]]++; for(int i=1;i<m;i++) c[i]+=c[i-1]; for(int i=n-1;i>=0;i--) sa[--c[x[i]]]=i; for(int j=1;j<=n;j<<=1) { p=0; for(int i=n-j;i<n;i++) y[p++]=i; for(int i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j; for(int i=0;i<m;i++) c[i]=0; for(int i=0;i<n;i++) c[x[y[i]]]++; for(int i=1;i<m;i++) c[i]+=c[i-1]; for(int i=n-1;i>=0;i--) sa[--c[x[y[i]]]]=y[i]; swap(x,y); p=1;x[sa[0]]=0; for(int i=1;i<n;i++) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; if(p>=n) break; m=p; } int k=0; n--; for(int i=0;i<=n;i++) ra[sa[i]]=i; for(int i=0;i<n;i++) { if(k) k--; int j=sa[ra[i]-1]; while(str[i+k]==str[j+k])k++; height[ra[i]]=k; } } int m,n,k,t; int ra[maxn],height[maxn],str[maxn],sa[maxn]; int mm[maxn],best[20][maxn]; void initRMQ(int n) { mm[0]=-1; for(int i=1;i<=n;i++) mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1]; for(int i=1;i<=n;i++)best[0][i]=height[i]; for(int i=1;i<=mm[n];i++) for(int j=1;j+(1<<i)-1<=n;j++) { int a=best[i-1][j]; int b=best[i-1][j+(1<<(i-1))]; if(a<b) best[i][j]=a; else best[i][j]=b; } } int askRMQ(int a,int b) { int t=mm[b-a+1]; b-=(1<<t)-1; a=best[t][a];b=best[t][b]; return a<b?a:b; } int lcp(int a,int b) { a=ra[a];b=ra[b]; if(a>b) swap(a,b); return askRMQ(a+1,b); } char s[maxn]; int main() { ///freopen("in.txt","r",stdin); int cas=0; while(scanf("%s",s)) { if(s[0]=='#') break; n=strlen(s); for(int i=0;i<n;i++) str[i]=s[i]-'a'+1; str[n]=0; da(str,sa,ra,height,n,28); /* for(int i=1; i<=n; ++i) { printf("%d:\t",i); for(int j=i-1; j<n; ++j) printf("%d",str[j]); puts(""); } puts(""); puts("-------------After sort---------------"); for(int i=1; i<=n; ++i) { printf("sa[%2d ] = %2d\t",i,sa[i]); for(int j=sa[i]; j<n; ++j)//for(int j=sa[i]; j<n; ++j)??? printf("%d",str[j]); puts(""); } puts(""); puts("---------------Height-----------------"); for(int i=1; i<=n; ++i) printf("height[%2d ]=%2d \n",i,height[i]); puts(""); puts("----------------ra------------------"); for(int i=1; i<=n; ++i) printf("ra[%2d ] = %2d\n",i,ra[i]); puts("------------------END-----------------"); */ initRMQ(n); int ans=1,l=0,r=0; for(int i=1;i<=n;i++) { for(int j=0;j+i<n;j+=i)//下面的程式碼上圖都有體現 { int len=lcp(j,j+i); int k=j-(i-len%i); int sum=len/i+1; //if(k>=0&&lcp(k,k+i)>=i) // sum++; int pl=j; for(int m=j+len%i;m>j-i&&m>=0&&str[m]==str[m+i];m--)//找字典序最小的 l { if(m==k) { sum++; pl=m; } else if(ra[m]<ra[pl]) { pl=m; } } if(ans<sum) { ans=sum; l=pl; r=l+sum*i-1; } else if(ans==sum&&ra[l]>ra[pl]) { l=pl; r=l+sum*i-1; } } } printf("Case %d: ",++cas); for(int i=l;i<=r;i++) printf("%c",s[i]); puts(""); } return 0; }