1. 程式人生 > >bzoj1717 [Usaco2006 Dec]Milk Patterns 產奶的模式

bzoj1717 [Usaco2006 Dec]Milk Patterns 產奶的模式

def mil cpp csdn algorithm code john oid 牛奶

Description

農夫 John 發現他的奶牛產奶的質量一直在變動。經過細致的調查,他發現:雖然他不能預見明天產奶的質量,但連續的若幹天的質量有很多重疊。我們稱之為一個“模式”。 John 的牛奶按質量可以被賦予一個 \(0\)\(1000000\) 之間的數。並且 John 記錄了 \(N(1\le N\le 20000)\) 天的牛奶質量值。他想知道最長的出現了至少 \(K(2\le K\le N)\) 次的模式的長度。比如 1 2 3 2 3 2 3 1 中 2 3 2 3 出現了兩次。當 \(K=2\) 時,這個長度為 \(4\)

Input

Line \(1\) : 兩個整數 \(N,K\)

Lines \(2\cdots N+1\) : 每行一個整數表示當天的質量值。

Output

Line \(1\) : 一個整數: \(N\) 天中最長的出現了至少 \(K\) 次的模式的長度

Sample Input

8 2
1
2
3
2
3
2
3
1

Sample Output

4

Solution

二分和後綴數組。具體看這兒吧

#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<cmath> using namespace std; #define N 2000001 #define rep(i, a, b) for (int i = a; i <= b; i++) #define drp(i, a, b) for (int i = a; i >= b; i--) #define ll long long inline int read() { int x = 0, flag = 1; char ch = getchar(); while (!isdigit(ch)) { if (!(ch ^ '-')) flag = -1
; ch = getchar(); } while (isdigit(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar(); return x * flag; } int n, m, K; char in[N]; int s[N]; int sa[N], t1[N], t2[N], c[N], rk[N], height[N]; void get_sa() { int *x = t1, *y = t2; rep(i, 1, n) c[x[i] = s[i]]++; rep(i, 1, m) c[i] += c[i - 1]; drp(i, n, 1) sa[c[x[i]]--] = i; for (int k = 1; k <= n; k <<= 1) { int p = 0; rep(i, n - k + 1, n) y[++p] = i; rep(i, 1, n) if (sa[i] > k) y[++p] = sa[i] - k; memset(c, 0, sizeof(c)); rep(i, 1, n) c[x[i]]++; rep(i, 1, m) c[i] += c[i - 1]; drp(i, n, 1) sa[c[x[y[i]]]--] = y[i]; swap(x, y), p = 0, x[sa[1]] = ++p; rep(i, 2, n) x[sa[i]] = y[sa[i - 1]] == y[sa[i]] && y[sa[i - 1] + k] == y[sa[i] + k] ? p : ++p; if ((m = p) == n) break; } } void get_height() { rep(i, 1, n) rk[sa[i]] = i; int k = 0; rep(i, 1, n) { if (k) k--; int p = sa[rk[i] - 1]; while (s[i + k] == s[p + k]) k++; height[rk[i]] = k; } } inline bool check(int x) { int cnt = 0; rep(i, 1, n) if (height[i] < x) cnt = 0; else if (++cnt == K - 1) return 1; return 0; } #define mid (l + r >> 1) int main() { n = read(), K = read(); rep(i, 1, n) s[i] = read() + 1, m = max(m, s[i]); get_sa(), get_height(); int l = 1, r = n, ans; while (l <= r) if (check(mid)) ans = mid, l = mid + 1; else r = mid - 1; printf("%d", ans); return 0; }

bzoj1717 [Usaco2006 Dec]Milk Patterns 產奶的模式