NC14301 K-th Number
K-th Number
https://ac.nowcoder.com/acm/problem/14301
NC14301 K-th Number
原题地址:
基本思路:
这题做法感觉还是蛮难想的,我也是参考大家的题解才想通,是一个二分+尺取。
我们先看题意,基本意思就是找数组A中每个长度大于K的区间里的第K大形成数组B,然后再在B中找第M大,看上去比较复杂。
我们看答案只是一个数,所以考虑二分答案,我们每次二分出这个第M大的值为X,那么我们在原数组中尺取左右端点L,R,如果在[L,R]范围内大于这个X的数出现大于等于K个那么就意味着这右端点之后的每个区间它的第K大一定也会是大于X的(这里是重点),知道了这一点左右尺取范围内大于X的数等于K的区间,每次通过右边界位置就能找出所有第K大大于X的区间的个数,并且这个个数是根据X的变大而变小的所以有了单调关系,那么这个二分就是成立的。
参考代码:
#pragma GCC optimize(2)
#pragma GCC optimize(3)
#include <bits/stdc++.h>
using namespace std;
#define IO std::ios::sync_with_stdio(false)
#define int long long
#define rep(i, l, r) for (int i = l; i <= r; i++)
#define per(i, l, r) for (int i = l; i >= r; i--)
#define mset(s, _) memset(s, _, sizeof(s))
#define pb push_back
#define pii pair <int, int>
#define mp(a, b) make_pair(a, b)
#define INF 0x3f3f3f3f
inline int read() {
int x = 0, neg = 1; char op = getchar();
while (!isdigit(op)) { if (op == '-') neg = -1; op = getchar(); }
while (isdigit(op)) { x = 10 * x + op - '0'; op = getchar(); }
return neg * x;
}
inline void print(int x) {
if (x < 0) { putchar('-'); x = -x; }
if (x >= 10) print(x / 10);
putchar(x % 10 + '0');
}
const int maxn = 1e5 + 10;
int n,k,m,a[maxn];
bool check(int x){
int j = 0,cnt = 0,res = 0;
rep(i,1,n){//左边界一直往右走;
while(j < n && cnt < k){//右边界尽量往后试探;
if(a[++j] >= x) cnt++;
}
if(cnt == k) res += n - j + 1;
if(a[i] >= x) cnt--;//左边界往后跑了,要清除原位置的影响;
}
return res >= m;
}
signed main(){
IO;
int t;
cin >> t;
while(t--){
cin >> n >> k >> m;
rep(i,1,n) cin >> a[i];
int l = 1,r = INF,ans = 0;
while(l <= r){//标准的二分;
int mid = (l + r) >> 1;
if(check(mid)) ans = mid,l = mid + 1;
else r = mid - 1;
}
cout << ans << '\n';
}
return 0;
}
查看8道真题和解析