1. 程式人生 > 實用技巧 >HDU - 6231

HDU - 6231

HDU - 6231

題意:

給你一個長度為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;

    }
}