L2 : 더 맵게 Python

jhyunn·2023년 1월 17일
0

Programmers

목록 보기
32/69

L2 : 더 맵게 Python

https://school.programmers.co.kr/learn/courses/30/lessons/42626

import heapq # 연속해서 sort할 일이 있으면 heapq가 더 빠름

def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville) # heapq 정렬
    while scoville[0]<K:
        if len(scoville)==1:
            return -1
        new = heapq.heappop(scoville) + heapq.heappop(scoville)*2 # 왼쪽(최솟값) pop
        heapq.heappush(scoville, new) # append & heapify
        answer += 1
    return answer

최솟값을 반복해서 찾아 연산해줘야한다.
이 경우, heapq를 활용하는 것이 속도에서 유리하다.

#heapq #heapify #heappop #heappush

profile
https://github.com/Sungjeonghyun

0개의 댓글