HDU 2459 Maximum repetition substring

题目：Maximum repetition substring

链接：http://acm.hdu.edu.cn/showproblem.php?pid=2459

题意：给你一个字符串，求连续重复出现次数最多的子串（不重叠），如果有多个，输出字典序最小的那个。别如aacdabcdab，a连续着重复出现2次，cdab连续着出现2次，但aa的字典序小，所以输出aa。

思路：

　　枚举+后缀树组+rmq

　　枚举重复串的长度，后缀数组预处理出height数组，rmq预处理height，实现O（1）查询。

　　给几组测试数据：

　　1. aabababa，答案为ababab，不是bababa

　　2. atbctbctb，答案是bctbct，不是ctbctb

　　3. aabcabcabcab，答案是abcabcabc，不是cabcabcab

 #include<stdio.h>
 #include<string.h>
 #include<math.h>
 #define N 200020
 char s1[];
 int ws[N],wv[N];
 int sa[N],r[N],wx[N],wy[N];
 int height[N];
 bool 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 n,int m)
 {
   //注意，这里的n必须比原始数组大小大1
   int *x=wx,*y=wy;
   for(int i=;i<m;i++) ws[i]=;
   for(int i=;i<n;i++) ws[x[i]=r[i]]++;
   for(int i=;i<m;i++) ws[i]+=ws[i-];
   for(int i=n-;i>=;i--) sa[--ws[x[i]]]=i;
   //这里的x[i] 表示下标i的第一关键字排名
   int i,j,p,*t;
   for(j=,p=;p<n;j*=,m=p)
   {
     for(p=,i=n-j;i<n;i++) y[p++]=i;
     for(i=;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
     //此时的y[i] 表示第二关键字排第i的下标是y[i]
     for(i=;i<n;i++) wv[i]=x[y[i]];
     for(i=;i<m;i++) ws[i]=;
     for(i=;i<n;i++) ws[wv[i]]++;
     for(i=;i<m;i++) ws[i]+=ws[i-];
     for(i=n-;i>=;i--) sa[--ws[wv[i]]]=y[i];
     for(t=x,x=y,y=t,p=,x[sa[]]=,i=;i<n;i++)
       x[sa[i]]=cmp(y,sa[i],sa[i-],j)?p-:p++;
   }
   for(int i=;i<n;i++)
   {
     r[sa[i]]=i;
   }
 }
 void calHeight(int n)
 {
   int h=;
   for(int i=;i<n;i++)
   {
     if(r[i]==) h=;
     else
     {
       int k=sa[r[i]-];
       if(--h<) h=;
       while(s1[k+h]==s1[i+h]) h++;
     }
     height[r[i]]=h;
   }
 }
 int f[][];
 int min(int x,int y)
 {
   return x<y?x:y;
 }
 int max(int x,int y)
 {
   return x<y?y:x;
 }
 void rmq(int n)
 {
   for(int j=;j<n;j++) f[j][]=height[j];
   for(int i=;i<;i++)
   {
     for(int j=;j<n;j++)
     {
       if(j+(<<i)- < n)
       {
         f[j][i]=min(f[j][i-],f[j+(<<i-)][i-]);
       }
     }
   }
 }
 int look(int x,int y)
 {
   x=r[x];
   y=r[y];
   if(x>y) x^=y^=x^=y;
   x++;
   int k=(int)log2((double)(y-x+));
   return min(f[x][k],f[y-(<<k)+][k]);
 }
 
 bool cmp(int sl,int l)
 {
   if(r[sl]>r[l]) return ;
   return ;
 }
 
 void solve(int n)
 {
   int maxt=,sl=,sr=;
   for(int i=;i<=n/;i++)
   {
     for(int j=;j+i<n;j+=i)
     {
       int l=j;
       int r=j+i;
       int lcp=look(l,r);
       int d=lcp/i+;
       int t=j-(i-lcp%i);
       for(int k=l-;k>=&&k+i>j&&s1[k]==s1[k+i];k--)
       {
         if(k==t)
         {
           d++;
           l=k;
         }
         else if(cmp(l,k))
         {
           l=k;
         }
       }
       if(d>maxt)
       {
         sl=l;
         sr=l+d*i-;
         maxt=d;
       }
       else if(d==maxt&&cmp(sl,l))
       {
         sl=l;
         sr=l+d*i-;
       }
     }
   }
   for(int i=sl;i<=sr;i++)
   {
     printf("%c",s1[i]);
   }
   printf("\n");
 }
 
 int main()
 {
   int cas=;
   while(scanf("%s",s1)!=EOF)
   {
     if(s1[]=='#') break;
     int len=strlen(s1);
     for(int i=;i<len;i++)
       r[i]=s1[i]-'a'+;
     r[len++]=;
     da(r,len,);
     calHeight(len);
     rmq(len);
     printf("Case %d: ",cas++);
     solve(len-);
 
   }
   return ;
 }