题目描述

模板题：第k个数

C++代码

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 1;

int quick_select(int a[], int l, int r, int k){
    if(l == r) //数组仅有一个数
        return a[l];
    int i = l - 1, j = r + 1, x = a[l + r >> 1];
    while(i < j){
        do i++; while(a[i] < x);
        do j--; while(a[j] > x);
        if(i < j) swap(a[i], a[j]);
    }
    int al = j - l + 1; //左侧 l~j 共有 al 个数
    if(al >= k){
        quick_select(a, l, j, k);
    }else{
        quick_select(a, j + 1, r, k - al);
    }
}

int main(){
    int arr[N];
    int n, k;
    cin >> n >> k;
    for(int i = 0; i < n; i++)
        cin >> arr[i];
    int res = quick_select(arr, 0, n - 1, k);
    cout << res << "\n";
    return 0;
}