HDU - 6231
阿新 • • 發佈:2020-11-07
題意:
給你一個長度為n的陣列, 將所有長度大於k的子串的第k大放入陣列B中, 然後詢問陣列B
的第m大是什麼?
題解:
考慮: 二分答案
如果我們知道當前答案為x, 那麼怎麼判斷x是第幾大呢, 可以在元素組裡面用兩個指標 \(l, r\)
列舉每個位置的\(l\) 然後走r當出現的大於等於x數字的個數大與 \(k\)的時候停止 走r, 那麼次使就會出現
\(n - r + 1\)個區間的第k大是大於等於x的, 然後在繼續列舉下一個l, 一次類推。
如果當前的答案最後出現大於等於x的區間大小等於m, 那麼就二分就想右分, 否則向左分
程式碼:
#include<bits/stdc++.h> using namespace std; const int N = 1e5 + 7; typedef long long ll; ll n, m, k, a[N]; bool judge(ll x) { ll r = 1; ll cnt = 0; ll ans = 0; for (int i = 1; i <= n; i++) { while (r <= n && cnt < k) { if (a[r] >= x) cnt++; r++; } if (cnt == k) ans += (n - r + 2); if (a[i] >= x) cnt--; } if (ans >= m) return true; return false; } int main() { ios::sync_with_stdio(0); int t; cin >> t; while (t--) { cin >> n >> k >> m; for (int i = 1; i <= n; i++) { cin >> a[i]; } ll l = 1, r = 1e9, ans; while (l <= r) { ll mid = (l + r) / 2; if (judge(mid)) { ans = mid; l = mid + 1; } else { r = mid - 1; } } cout << ans << endl; } }