11004번: K번째 수

https://www.acmicpc.net/problem/11004


$O(n)$

빠른 selection algorithm을 구현하거나 nth_element 함수를 사용하자.

#include<cstdio>
#include<algorithm>
int n, k, a[5000000];
int main() {
    scanf("%d %d", &n, &k);
    for (int i = 0; i < n; i++) scanf("%d", a + i);
    std::nth_element(a, a + k - 1, a + n);
    printf("%d", a[k - 1]);
    return 0;
}