K-th Number

K-th Number

https://ac.nowcoder.com/acm/problem/14301

看直播时似乎很简单,但实际做题时,细节很烦人。
题意:我们要取最终添加完成的数组中第M打的元素
利用二分的思想,这个答案肯定在1 到 1e9之间,那么我们在这个范围内进行二分最终获得答案
那么具体怎么操作呢?
假如我们二分枚举了一个数,我们需要知道在最终排好序的数组中比它大的一共有几个数。
如果比他大或等于他的数大于等于M那么我们可能取小了,也可能最终答案就是他。
如果比他大或等于他的数小于M那么我们一定是取大了。
那么对于我们可能取小也可能取到正确答案的数,也就是第一种情况我们令l = mid+1,进行放大。
对于我们一定取大的数,令r = mid-1,进行缩小。
利用check()函数检查大于等于他的数是否大于等于m
这样最终我们一定会得到正确答案!
正确答案是l-1,也就是最后一次check()为true的mid
假设,我们这一次取到了正确答案,那么按照规则会会进行放大,接下来的每一个数check()都会为false
那么最终l不会变化,答案就是l-1。
那么check(int num)1函数该如何验证大于等于num的数是否大于等于M呢?
对于一个区间[a1,a2,a3,a4,a5,a6,a7,a8]如果,其中大于等于num的数大于等于k那么
这个区间的第k大元素一定也是大于等于k的.这是充要条件!!!!
而如果这个区间中大于等于num的数都大于等于k了,那么我们扩大这个区间的右边界,所得到的区间
也一定满足条件!
那么,我们尺取法,从头到尾进行线性扫描,得出答案。
复杂度O(nlogn)
细节要注意!

#include<iostream>;
#include<algorithm>;
using namespace std;
typedef long long ll;
const int max_n = 1e5 + 100;
ll n, k, m;
ll a[max_n];

bool check(ll x) {
    ll cnt = 0;ll ans = 0;
    for (int i = 1, j = 1;j <= n;j++) {
        if (a[j] >= x)cnt++;
        while (cnt >= k) {
            ans += n + 1 - j;
            if (a[i] >= x)cnt--;
            i++;
        }
    }return ans >= m;
}

ll solve() {
    ll r = 0;ll l = a[1];
    for (int i = 1;i <= n;i++)
        r = max(r, a[i]), l = min(l, a[i]);
    while (l <= r) {
        ll mid = (l + r) >> 1;
        if (check(mid))l = mid + 1;
        else r = mid - 1;
    }return l - 1;
}

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];
        cout << solve() << endl;
    }
}
全部评论

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务