Milk Patterns POJ

Milk Patterns

 POJ - 3261

题意：问长度为n的串中至少出现k次的子串最长是多长。

二分长度，再判断是否有子串重复k次。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ull unsigned long long
const int maxn=20010;
const int seed=131;

ull base[maxn],h[maxn];
int a[maxn];
int n,k;

int check(int m)
{
    memset(h,0,sizeof(h));
    for(int i=0;i<m;i++) h[m-1]=h[m-1]*seed+a[i];
    for(int i=m;i<n;i++){
        h[i]=h[i-1]*seed-a[i-m]*base[m]+a[i];
    }
    sort(h,h+n);
    int cnt=1;
    for(int i=n-1;i>=m;i--){
        if(h[i]==h[i-1]) cnt++;
        else cnt=1;
        if(cnt>=k) return 1;
    }
    return 0;
}
int main()
{
    base[0]=1;
    for(int i=1;i<maxn;i++) base[i]=base[i-1]*seed;
    while(scanf("%d%d",&n,&k)!=EOF){
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        int L=0,R=n;
        while(L<=R){
            int m=L+(R-L)/2;
            if(check(m)) L=m+1;
            else R=m-1;
        }
        printf("%d
",R);
    }
}