[#134][알고리즘][백준] 11004. K번째 수

[백준] 11004. K번째 수

문제 링크(https://www.acmicpc.net/problem/11004)

문제

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

입력

첫째 줄에 N(1 ≤ N ≤ 5,000,000)과 K (1 ≤ K ≤ N)이 주어진다.
둘째에는 A1, A2, ..., AN이 주어진다. (-109 ≤ Ai ≤ 109)

출력

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

예제 입력 1 

5 2
4 1 2 3 5

예제 출력 1 

2
C++풀이
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
 
int main() {
    int N, K;
    //int *arr;
    vector<int> input;
    cin >> N >> K;
    //arr = new int[N];
    input.resize(N);
    for (int i = 0; i < N; i++) {
        scanf("%d"&input[i]);
    }
 
    sort(input.begin(), input.end());
 
    printf("%d", input[K - 1]);
    return 0;
}
 
cs
처음에 Vector로 풀었는데 시간초과가 나서.. 당황...
배열로 바꿨더니 정답이래서 뭔가 이상해서 다시 Vector로 수정해봤는데 정답처리됬답..
이상이상..

댓글

가장 많이 본 글