문제
수 N개 A1, A2, ..., AN이 주어진다. A를 오름차순 정렬했을 때, 앞에서부터 K번째 있는 수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 N(1 ≤ N ≤ 5,000,000)과 K (1 ≤ K ≤ N)이 주어진다.
둘째에는 A1, A2, ..., AN이 주어진다. (-109 ≤ Ai ≤ 109)
출력
A를 정렬했을 때, 앞에서부터 K번째 있는 수를 출력한다.
주어진 수를 정렬하고 K번째 수를 출력하는 간단한 문제이다.
예제 입출력은 다음과 같다.
CASE | INPUT | OUTPUT |
1 | 5 2 4 1 2 3 5 |
2 |
작성 코드
n, k = map(int, input().split())
array = list(map(int, input().split()))
array.sort()
print(array[k-1])
별로 설명할 게 없네유
근데 K의 범위가 (1<= K <=N)이면 K가 인덱스가 아니라 말 그대로 입력된 A 중에 몇 번째 수인지를 찾으라는 건데 예제 입출력이랑은 다르다.
예제를 보면 2번째 수는 1인데 출력을 보면 3번째 수가 출력된 것으로 보아 입력된 K를 인덱스로 생각해야 하는게 맞다.
조금 이상허다..
'코딩 테스트 > Baekjoon' 카테고리의 다른 글
[백준] 1568. 새 (0) | 2021.04.14 |
---|---|
[백준] 1543. 문서 검색 (0) | 2021.04.14 |
[백준] 2751. 수 정렬하기 2 (0) | 2021.04.13 |
[백준] 7490. 0 만들기 (0) | 2021.04.13 |
[백준] 1874. 스택 수열 (0) | 2021.04.12 |