1717: [Usaco2006 Dec]Milk Patterns 产奶的模式

1717: [Usaco2006 Dec]Milk Patterns 产奶的模式

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 1469  Solved: 799
[Submit][Status][Discuss]

Description

农夫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。

Input

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

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

Output

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

Sample Input

8 2
1
2
3
2
3
2
3
1

Sample Output

4

题意

给定一个字符串，求至少出现k次的可重叠最长重复子串。

分析

后缀数组求出height（排名相邻的两个后缀的公共前缀），二分长度，在height中O(n)判断。

要求有连续k-1个height的值大于等于当前二分的长度才可以。

时间复杂度O(nlogn)

直接二分+hash也可以，复杂度相同

code

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 50010;

int s[N];
int t1[N],t2[N],c[N],sa[N],height[N],rank[N];
int n,m = 130,k;

void get_sa() {
    int i,p,*x = t1,*y = t2;
    for (i=0; i<m; ++i) c[i] = 0;
    for (i=0; i<n; ++i) x[i] = s[i],c[x[i]]++;
    for (i=1; i<m; ++i) c[i] += c[i-1];
    for (i=n-1; i>=0; --i) sa[--c[x[i]]] = i;
    for (int k=1; k<=n; k<<=1) {
        p = 0;
        for (i=n-k; i<n; ++i) y[p++] = i;
        for (i=0; i<n; ++i) if(sa[i]>=k) y[p++]=sa[i]-k;
        for (i=0; i<m; ++i) c[i] = 0;
        for (i=0; i<n; ++i) c[ x[y[i]] ]++;
        for (i=1; i<m; ++i) c[i] += c[i-1];
        for (i=n-1; i>=0; --i) sa[--c[ x[y[i]] ]] = y[i];
        swap(x,y);
        p = 1;
        x[sa[0]] = 0;
        for (i=1; i<n; ++i) 
            x[sa[i]] = (y[sa[i-1]]==y[sa[i]] && sa[i-1]+k<n && sa[i]+k<n &&    
            y[sa[i-1]+k]==y[sa[i]+k]) ? p-1 : p++;        
        if (p>=n) break;
        m = p;
    }
}
void get_height() {
    for (int i=0; i<n; ++i) rank[sa[i]] = i;
    int k = 0;
    height[0] = 0;
    for (int i=0; i<n; ++i) {
        if (!rank[i]) continue;
        if (k) k--;
        int j = sa[rank[i]-1];
        while (i+k<n && j+k<n && s[i+k]==s[j+k]) k++;
        height[rank[i]] = k;
    }
}
bool check(int x) {
    int num = 0;
    for (int i=0; i<n; ++i) {
        if (height[i] >= x) {
            num ++;
            if (num == k-1) return true;
        }
        else num = 0;
    }
    return false;
}
int main () {
    scanf("%d%d",&n,&k);
    for (int i=0; i<n; ++i) scanf("%d",&s[i]);
    get_sa();
    get_height();
    int L = 1,R = n,mid,ans;
    while (L <= R) {
        mid = (L + R) >> 1;
        if (check(mid)) ans = mid,L = mid + 1;
        else R = mid - 1;
    }
    printf("%d",ans);
    return 0;
}