第k个数

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e6+10;

int n, a[N];
int k;

inline void quick_sort(int a[], int l, int r) {
    if(l >= r) return ;
    int x = a[(l+r)>>1], i = l-1, j = r+1;
    while(i < j) {
        while(a[++i] < x) ;
        while(a[--j] > x) ;
        if(i < j) swap(a[i], a[j]);
    }
    quick_sort(a, l, i-1);
    quick_sort(a, j+1, r);
}

priority_queue <int, vector <int>, less<int>> q;

inline void solve() {
    for(int i = 1; i <= n; i++) {
        q.push(a[i]);
        if(q.size() > k) q.pop();
    }
    cout << q.top() << '\n';
}

int main(int argc, char **argv) {
    cin.tie(0)->sync_with_stdio(false);
    
    cin >> n >> k;
    for(int i = 1; i <= n; i++) cin >> a[i];
    quick_sort(a, 1, n);
    // sort(a+1, a+1+n);
    // nth_element(a+1, a+k, a+1+n);
    cout << a[k] << '\n';
    
    return 0;
}
全部评论

相关推荐

05-29 09:02
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务