后缀数组练习2：可重叠的k次最长重复子串

其实和上一题是差不多的，只是在二分check的时候有一些小小的改动

1468: 后缀数组2：可重叠的k次最长重复子串

poj3261

时间限制: 1 Sec  内存限制: 128 MB
提交: 113  解决: 48
[提交] [状态] [讨论版] [命题人:admin]

题目描述

【问题描述】

农夫John发现他的奶牛产奶的质量一直在变动。经过细致的调查，他发现：虽然他不能预见明天 产奶的质量，但连续的若干天的质量有很多重叠。我们称之为一个“模式”。 John的牛奶按质量可以被赋予一个0到1000000之间的数。并且John记录了N(1<=N<=20000)天的 牛奶质量值。他想知道最长的出现了至少K(2<=K<=N)次的模式的长度。 比如1 2 3 2 3 2 3 1 中 2 3 2 3出现了两次。当K=2时，这个长度为4。（可重叠的k次最长重复子串）

【输入格式】

* Line 1: 两个整数 N,K。

* Lines 2..N+1: 每行一个整数表示当天的质量值。

（多组数据）

【输出格式】

* Line 1: 一个整数：N天中最长的出现了至少K次的模式的长度

【样例】
输入：

8 2

1

2

3

2

3

2

3

1
输出：

4

做法是是和上一题：不可重叠的最长重复子串是差不多的，就是要判断后缀分成若干组，然后判断有没有一个组后缀个数是不小于k的，存在就满足，不存在就不满足

代码实现，没什么注释，所以我就只放一个版本

 /*后将后缀分成若干组。 不同的是，这里要判断的是有没有一个组的后缀个数不小于 k。
 如果有，那么存在k 个相同的子串满足条件，否则不存在*/
 #include<cstdio>
 #include<cstring>
 #include<cstdlib>
 #include<algorithm>
 #include<cmath>
 #include<iostream>
 using namespace std;
 int sa[],Rank[],rsort[];
 int a[],cnt[],pos[],height[];
 bool cmp(int x,int y,int k){return cnt[x]==cnt[y] && cnt[x+k]==cnt[y+k];}
 void get_sa(int n,int m)
 {
     int k=,p=,len;
     for(int i=;i<=n;i++) Rank[i]=a[i];
     memset(rsort,,sizeof(rsort));
     for(int i=;i<=n;i++) rsort[Rank[i]]++;
     for(int i=;i<=m;i++) rsort[i]+=rsort[i-];
     for(int i=n;i>=;i--) sa[rsort[Rank[i]]--]=i;
     for(int k=;k<n;k<<=)
     {
         len=;
         for(int i=n-k+;i<=n;i++) pos[++len]=i;
         for(int i=;i<=n;i++) if(sa[i]>k) pos[++len]=sa[i]-k;
         for(int i=;i<=n;i++) cnt[i]=Rank[pos[i]];
         memset(rsort,,sizeof(rsort));
         for(int i=;i<=n;i++) rsort[cnt[i]]++;
         for(int i=;i<=m;i++) rsort[i]+=rsort[i-];
         for(int i=n;i>=;i--) sa[rsort[cnt[i]]--]=pos[i];
         for(int i=;i<=n;i++) cnt[i]=Rank[i];
         p=; Rank[sa[]]=;
         for(int i=;i<=n;i++)
         {
             if(!cmp(sa[i],sa[i-],k)) p++;
             Rank[sa[i]]=p;
         }
         if(p==n) break; m=p;
     }
     a[]=; sa[]=;
 }
 void get_he(int n)
 {
     int j,k=;
     for(int i=;i<=n;i++)
     {
         j=sa[Rank[i]-];
         if(k) k--;
         while(a[j+k]==a[i+k]) k++;
         height[Rank[i]]=k;
     }
 }
 bool check(int mid,int n,int k)
 {
     int sum=;
     for(int i=;i<=n;i++)
     {
         if(height[i]>=mid)/*满足条件*/
         {
             sum++;/*个数增加*/
             if(sum==k) return true;/*如果满足条件就直接返回*/
         }
         else sum=;/*否则重新找*/
     }
     return false;
 }
 int main()
 {
     int n,k;
     while(scanf("%d%d",&n,&k)!=EOF)
     {
         for(int i=;i<=n;i++) scanf("%d",&a[i]);
         get_sa(n,); get_he(n);
         int l=,r=n,ans=;
         while(l<=r)
         {
             int mid=(l+r)/;
             if(check(mid,n,k))
             {
                 ans=mid;
                 l=mid+;
             }
             else r=mid-;
         }
         printf("%d\n",ans);
     }
     return ;
 }

Tristan Code