POJ 3693 Maximum Repetition Substring <后缀数组 + RMQ>

来源:互联网 发布:网络舆情是什么 编辑:程序博客网 时间:2024/05/17 22:51

题目:传送门。


分析:

求连续重复次数最多的子串,若有多个,输出字典序最小的。感觉有点难度了,看了解题报告做的。


代码:

#include <cstdio>#include <cstring>#include <algorithm>#include <cmath>using namespace std;const int MAXN=100005;const int INF=0x3f3f3f3f;int wa[MAXN],wb[MAXN],wv[MAXN],w[MAXN],height[MAXN];int sa[MAXN];int *rank;char str[MAXN];int data[MAXN];int dp[MAXN][17];bool cmp(int *r,int a,int b,int l){      return r[a]==r[b]&&r[a+l]==r[b+l];  }    void da(int n,int m){      int *x=wa,*y=wb,*t,p=1;      for(int i=0;i<m;++i) w[i]=0;      for(int i=0;i<n;++i) ++w[x[i]=data[i]];      for(int i=1;i<m;++i) w[i]+=w[i-1];      for(int i=n-1;i>=0;--i) sa[--w[x[i]]]=i;      for(int len=1;p<n;len<<=1,m=p){          p=0;          for(int i=n-len;i<n;++i) y[p++]=i;          for(int i=0;i<n;++i) if(sa[i]>=len) y[p++]=sa[i]-len;          for(int i=0;i<n;++i) wv[i]=x[y[i]];          for(int i=0;i<m;++i) w[i]=0;          for(int i=0;i<n;++i) ++w[wv[i]];          for(int i=1;i<m;++i) w[i]+=w[i-1];          for(int i=n-1;i>=0;--i) sa[--w[wv[i]]]=y[i];          t=x; x=y; y=t; p=1; x[sa[0]]=0;          for(int i=1;i<n;++i) x[sa[i]]=cmp(y,sa[i-1],sa[i],len)?p-1:p++;      }      rank=x;  }    void calHeight(int n){      for(int i=0,k=0,j;i<n;height[rank[i++]]=k)          for(k?--k:k,j=sa[rank[i]-1];data[i+k]==data[j+k];++k);  }void initRMQ(int n){int l=(int)(log(n*1.0)/log(2.0));for(int i=1;i<=n;++i) dp[i][0]=height[i];for(int j=1;j<=l;++j){for(int i=1;i+(1<<j)-1<=n;++i){dp[i][j]=min(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);}}}int LCP_Inquiry(int a,int b){int i=rank[a],j=rank[b];if(i>j) swap(i,j); ++i;int t=(int)((log(j-i+1.0)/log(2.0)));return min(dp[i][t],dp[j-(1<<t)+1][t]);}int main(){int cas=0;while(scanf("%s",str) && str[0]!='#'){int len=strlen(str),m=0;for(int i=0;i<len;++i){data[i]=str[i]-'a'+1;if(m<data[i]) m=data[i];}data[len]=0;da(len+1,m+2);calHeight(len);initRMQ(len);int max_t=1,star=sa[1],max_len=1;;for(int l=1;l<=len/2;++l){for(int i=0;i+l<len;i+=l){if(str[i]!=str[i+l]) continue;int t=i+l+LCP_Inquiry(i,i+l)-1;int j=i;while(j>=0&&j>i-l&&str[j]==str[j+l]){int rp=(t-j+1)/l;if(rp>max_t || (rp==max_t && rank[j]<rank[star])){star=j;max_len=l;max_t=rp;}--j;}}}str[star+max_t*max_len]='\0';printf("Case %d: ",++cas);printf("%s\n",&str[star]);}return 0;}


0 0