2017CCPC 哈尔滨 B

 

这题没有考虑到m这个东西，所以就没有往二分答案的方向想

 

二分答案

 

check的时候，我们找的是大于等于x的数有多少个被加入到那个数组中。如果 >= m说明这个数可能是答案，否则就不是。

 

用尺取来计数，还算比较好理解。

 

 

#include <bits/stdc++.h>

using namespace std;

int n,k;
long long m;

int a[100005];
int b[100005];

bool check(int x) {

    long long ans = 0;
    int num = 0;
    int pos = 1;

    for(int i = 1; i <= n; i++) {
        if(a[i] >= x) {
            num++;
        }
        if(num == k) {

            // 后面形成的区间，x作为第k大，或者有比它大的
            // 但是一定会有n - i + 1个大于等于x的数被加入到ans数组中
            ans += n - i + 1;

            // 如果左端这个数小于x
            // 那么去掉他没有影响
            // 依然会有n - i + 1个大于等于x的数加入到ans数组中
            while(a[pos] < x) {
                ans += n - i + 1;
                pos++;
            }
            pos++;
            num--;
        }


    }

    //cout<<ans<<endl;
    if(ans >= m) return true;
    return false;
}

int main() {

    int t;
    scanf("%d",&t);

    while(t--) {
        scanf("%d %d %lld",&n,&k,&m);

        for(int i = 1; i <= n; i++) {
            scanf("%d",&a[i]);
        }

        //cout<<num<<endl;
        int L = 1,R = 1000000001;
        int mid;
        while(R - L > 1) {
            mid = L + R >> 1;

            //cout<<mid<<' '<<b[mid]<<endl;
            if(check(mid)) {
                L = mid;
            }
            else {
                R = mid;
            }
        }


        printf("%d\n",L);

    }





    return 0;
}