16455. K번째 수 찾는 함수

c
#include <bits/stdc++.h>

int kth(std::vector<int> &a, int k) {
    nth_element(a.begin(), a.begin() + k - 1, a.end());
    return a[k - 1];
}