heapq로 k번째 최솟값 - 파이썬 실습

LEARNING GUIDE

📖 학습 내용

학습 내용

정수 여러 개와 k가 한 줄씩 입력됩니다. k번째로 작은 값을 힙으로 구해 출력하세요.

📥 테스트 입력값

5 3 8 1 9 2
3

🔎 실행 결과

💡 힌트 코치

heapq.nsmallest(k, nums)[-1] 을 사용하세요.

✅ 정답 공개

import heapq
nums = list(map(int, input().split()))
k = int(input())
print(heapq.nsmallest(k, nums)[-1])

◀ 이전 목록 다음 ▶