K-th Number 二分

题目链接

题意

n个元素的数组A，从A数组中所有长度大于等于k的区间中取出第k大放入B数组中，求B数组的第m大。
思路
这题正常想很难搞，看了题解才懂。mark一下。
考虑二分的做法，问题是求B数组第m大的是谁，假设为ans，考虑一个值x，求B数组里大于等于x的值是否大于m个，这个问题具有单调性，即求A数组里区间第k大的数大于等于x的区间是否多于m个。然而这个问题也不好解，神犇的做法来了，将数组A里大于等于x的值赋为1，否则为0，只要一个区间的和大于等于k，那么这个区间第k大的数就大于等于x，然后就是求有多少个区间和大于等于k了，考虑固定了区间右端点j，当左端点为i，[i,j]的和大于等于k时，那么左端点小于i的也肯定满足条件，那么只要尺取法扫一遍数组就行。最后判断区间和大于等于k的个数是不是多于m就行。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 7;
typedef long long ll;
int n, k, a[maxn], sum[maxn];
ll m;
bool check(int x) {
    ll ans = 0;
    for (int i = 1; i <= n; i++) sum[i] = sum[i - 1] + (a[i] >= x);
    int last = 0;
    for (int i = k; i <= n; i++) {
        while (sum[i] - sum[last] >= k) last++;
        ans += last;
    }
    return ans >= m;
}
int main()
{
    int t;
    scanf("%d", &t);
    while (t--) {
        scanf("%d%d%lld", &n, &k, &m);
        int left = 1, right = 1, ans;
        for (int i = 1; i <= n; i++) scanf("%d", &a[i]), left = min(left, a[i]), right = max(a[i], right);
        while (left <= right) {
            int mid = (left + right) / 2;
            if(check(mid)) {
                ans = mid;
                left = mid + 1;
            }
            else right = mid - 1;
        }
        printf("%d\n", ans);
    }
}