后缀数组练习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[1000010],Rank[1000010],rsort[1000010];
int a[1000010],cnt[1000010],pos[100010],height[1000010];
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=1,p=0,len;
    for(int i=1;i<=n;i++) Rank[i]=a[i];
    memset(rsort,0,sizeof(rsort));
    for(int i=1;i<=n;i++) rsort[Rank[i]]++;
    for(int i=1;i<=m;i++) rsort[i]+=rsort[i-1];
    for(int i=n;i>=1;i--) sa[rsort[Rank[i]]--]=i;
    for(int k=1;k<n;k<<=1)
    {
        len=0;
        for(int i=n-k+1;i<=n;i++) pos[++len]=i;
        for(int i=1;i<=n;i++) if(sa[i]>k) pos[++len]=sa[i]-k;
        for(int i=1;i<=n;i++) cnt[i]=Rank[pos[i]];
        memset(rsort,0,sizeof(rsort));
        for(int i=1;i<=n;i++) rsort[cnt[i]]++;
        for(int i=1;i<=m;i++) rsort[i]+=rsort[i-1];
        for(int i=n;i>=1;i--) sa[rsort[cnt[i]]--]=pos[i];
        for(int i=1;i<=n;i++) cnt[i]=Rank[i];
        p=1; Rank[sa[1]]=1;
        for(int i=2;i<=n;i++)
        {
            if(!cmp(sa[i],sa[i-1],k)) p++;
            Rank[sa[i]]=p;
        }
        if(p==n) break; m=p;
    }
    a[0]=0; sa[0]=0;
}
void get_he(int n)
{
    int j,k=0;
    for(int i=1;i<=n;i++)
    {
        j=sa[Rank[i]-1];
        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=1;
    for(int i=2;i<=n;i++)
    {
        if(height[i]>=mid)/*满足条件*/
        {
            sum++;/*个数增加*/
            if(sum==k) return true;/*如果满足条件就直接返回*/
        }
        else sum=1;/*否则重新找*/
    }
    return false;
}
int main() 
{
    int n,k;
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        get_sa(n,1000010); get_he(n);
        int l=0,r=n,ans=0;
        while(l<=r)
        {
            int mid=(l+r)/2;
            if(check(mid,n,k))
            {
                ans=mid;
                l=mid+1;
            }
            else r=mid-1;
        }
        printf("%d
",ans);
    }
    return 0;
}