BZOJ 1717: [Usaco2006 Dec]Milk Patterns 产奶的模式( 二分答案 + 后缀数组 )

时间:2023-03-08 17:48:53

BZOJ 1717: [Usaco2006 Dec]Milk Patterns 产奶的模式( 二分答案 + 后缀数组 )

二分答案m, 后缀数组求出height数组后分组来判断.

------------------------------------------------------------

#include<bits/stdc++.h>
using namespace std;
const int maxn = 20009;
struct HASH {
int id[maxn], N;
HASH() {
N = 0;
}
inline void work() {
sort(id, id + N);
N = unique(id, id + N) - id;
}
inline void add(int x) {
id[N++] = x;
}
inline int hash(int x) {
return lower_bound(id, id + N, x) - id;
}
} h;
int sa[maxn], rank[maxn], height[maxn], S[maxn], cnt[maxn], N, K;
void build_sa() {
int m = h.N, *x = height, *y = rank;
for(int i = 0; i < m; i++) cnt[i] = 0;
for(int i = 0; i < N; i++) cnt[x[i] = S[i]]++;
for(int i = 1; i < m; i++) cnt[i] += cnt[i - 1];
for(int i = N - 1; ~i; i--) sa[--cnt[x[i]]] = i;
for(int k = 1; k <= N; k <<= 1) {
int p = 0;
for(int i = N - k; i < N; i++) y[p++] = i;
for(int i = 0; i < N; i++) if(sa[i] >= k) y[p++] = sa[i] - k;
for(int i = 0; i < m; i++) cnt[i] = 0;
for(int i = 0; i < N; i++) cnt[x[i]]++;
for(int i = 1; i < m; i++) cnt[i] += cnt[i - 1];
for(int i = N - 1; ~i; i--) sa[--cnt[x[y[i]]]] = y[i];
swap(x, y); p = 1; x[sa[0]] = 0;
for(int i = 1; i < N; i++)
   x[sa[i]] = y[sa[i - 1]] == y[sa[i]] && y[sa[i - 1] + k] == y[sa[i] + k] ? p - 1 : p++;
if(p >= N) break;
m = p;
}
}
void build_height() {
int k = 0;
for(int i = 0; i < N; i++) rank[sa[i]] = i;
for(int i = 0; i < N; i++) {
if(k) k--;
int j = sa[rank[i] - 1];
while(S[i + k] == S[j + k]) k++;
height[rank[i]] = k;
}
}
bool check(int len) {
int cnt = 1;
for(int i = 1; i < N; i++) {
if(height[i] >= len) {
   if(++cnt >= K) return true;
} else
   cnt = 1;
}
return false;
}
int main() {
scanf("%d%d", &N, &K);
for(int i = 0; i < N; i++) {
   scanf("%d", S + i);
   h.add(S[i]);
}
h.add(S[N++] = -0x7fffffff);
h.work();
for(int i = 0; i < N; i++)
   S[i] = h.hash(S[i]);
build_sa(); build_height();
int L = 0, R = N;
    while(L < R) {
int m = (L + R) / 2 + 1;
if(check(m))
   L = m;
else 
   R = m - 1;
}
printf("%d\n", L);
return 0;
}

------------------------------------------------------------

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

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 640  Solved: 358
[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

HINT

Source