7230 : K번째 원소
•시간제한: 2.000 S
•메모리제한: 256 MB
•제출: 0
•통과: 0
•만든사람: suakii
문제 설명
$N$개의 정수가 주어질 때, 정렬했을 때 앞에서 $K$번째에 위치하는 원소의 값을 출력하십시오.
[제약 조건]
• 입력되는 모든 수는 정수입니다.
heapq.nsmallest() 또는 정렬을 사용하여 구현할 수 있습니다.[제약 조건]
• 입력되는 모든 수는 정수입니다.
입력 설명
첫째 줄에 $N$과 $K$가 주어집니다. ($1 \le K \le N \le 100{,}000$)
둘째 줄에 $N$개의 정수가 주어집니다.
둘째 줄에 $N$개의 정수가 주어집니다.
출력 설명
정렬했을 때 앞에서부터 $K$번째에 위치하는 원소의 값을 출력합니다.
입력 예시
COPY
복사되었습니다!
복사되었습니다!
5 3
3 1 4 1 5
출력 예시
COPY
복사되었습니다!
복사되었습니다!
3도움
sorted(arr)[K-1]로 간단히 구할 수 있습니다. 더 효율적인 방법으로는 heapq.nsmallest(K, arr)[-1]을 사용합니다.