[모의2-6] 최소 힙으로 K 합계

LEARNING GUIDE

📖 학습 내용

학습 내용

정수 여러 개와 k가 한 줄씩 입력됩니다. 가장 작은 k개의 합을 출력하세요.

📥 테스트 입력값

5 3 8 1 9 2 7
3

🔎 실행 결과

💡 힌트 코치

heapq.nsmallest 를 사용하세요.

✅ 정답 공개

import heapq
nums=list(map(int,input().split()))
k=int(input())
print(sum(heapq.nsmallest(k,nums)))

◀ 이전 목록 다음 ▶

PYTHON 3 EDITOR ● READY

실행 결과

INPUT (입력값) SMART MODE

OUTPUT (출력값)

코드를 실행하면 결과가 여기에 나타납니다.