반응형
programmers.co.kr/learn/courses/30/lessons/42626
import heapq
def solution(scoville, K):
cnt = 0
heapq.heapify(scoville)
while(1):
min1 = heapq.heappop(scoville)
if(min1<K and len(scoville)>=1):
min2 = heapq.heappop(scoville)
heapq.heappush(scoville,min1+min2*2)
cnt+=1
else: break
if(scoville[0]>=K): return cnt
else: return -1
정렬보다 더 빠른 알고리즘으로 힙을 사용했다.
반응형
'코딩 테스트 > 프로그래머스' 카테고리의 다른 글
[프로그래머스 Python] - 괄호변환(2020 KAKAO BLIND RECRUITMENT) (0) | 2020.11.23 |
---|---|
[프로그래머스 Python] - 전화번호 목록 (0) | 2020.10.30 |
[프로그래머스 Python] - 완주하지 못한 선수 (0) | 2020.10.30 |
프로그래머스 - 예산[Python] (0) | 2020.07.07 |
프로그래머스 - 가장 먼 노드[Python] (0) | 2020.07.06 |