[백준] 11004번: K번째 수 - C++

문제

수 N개 A₁, A₂, ..., A_N이 주어진다. A를 오름차순 정렬했을 때, 앞에서부터 K번째 있는 수를 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N(1 ≤ N ≤ 5,000,000)과 K (1 ≤ K ≤ N)이 주어진다.

둘째에는 A₁, A₂, ..., A_N이 주어진다. (-10⁹ ≤ A_i ≤ 10⁹)

 

출력

A를 정렬했을 때, 앞에서부터 K번째 있는 수를 출력한다.

---

정렬을 이용하여 풀 수 있는 문제.

수를 입력을 받은 뒤, sort 함수를 이용하여 정렬한 뒤 K번째 수를 출력하면 되는 간단한 문제다.

 

코드

#include <cstdio>
#include <algorithm>
#include <vector>

using namespace std;

int main(void) {
    int n, k;
    scanf("%d %d", &n, &k);
    vector<int> num(n);
    
    for(int i = 0; i < n; i++) {
        scanf("%d", &num[i]);
    }

    sort(num.begin(), num.end());
    printf("%d\n", num[k - 1]);
    return 0;
}