牛客 K-th Number（二分答案+验证）

K-th Number

题目链接

题意：
对数列A的每个长度大于等于K的区间取其第K大的数，插入到B中，再求B的第M大数。

code

// 对数列A的每个长度大于等于K的区间取其第K大的数，插入到B中，再求B的第M大数

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 5;
ll n, k, m;
int a[N];
bool check(ll x) // B中第M大的数是x，查找并判断A中有多少个区间的第K大数大于等于x
{
    ll l = 1, r = 0, cnt = 0, res = 0; // L、R双指针表示闭区间 [L,R] ，cnt 表示大于等于 x 的数有几个，res 记录至少有K个大于等于x的数的区间个数
    while (l <= n)
    {
        while (r < n && cnt < k) // 找右临界R
            if (a[++r] >= x)     // ++r
                cnt++;
        if (cnt == k)
            res += n - r + 1; // 区间 [L,R] 中有K个大于等于x的数，那么R右边的所有区间都至少有K个
        if (a[l++] >= x)      // 左界缩小一个 l++
            cnt--;
    }
    return res >= m;
}
int main()
{
#ifdef LZH_LOCAL
    freopen("in.in", "r", stdin);
    // freopen("out.out", "w", stdout);
#endif
    int T;
    cin >> T;
    while (T--)
    {
        cin >> n >> k >> m;
        for (int i = 1; i <= n; i++)
            cin >> a[i];
        ll l = 0, r = 1e9;
        while (l <= r)
        {
            ll mid = (l + r) >> 1; // 二分答案
            if (check(mid))
                l = mid + 1;
            else
                r = mid - 1;
        }
        cout << r << endl;
    }
    return 0;
}