[Programmers] 더 맵게

태환·2024년 3월 23일
0

Coding Test

목록 보기
143/151

📌 [Programmers] 더 맵게

📖 문제

📖 예제

📖 풀이

def solution(scoville, K):
    import heapq
    
    answer = 0
    heapq.heapify(scoville)
    
    while scoville:
        if scoville[0] >= K:
            return answer
        if len(scoville) == 1:
            return -1
        new = heapq.heappop(scoville) + heapq.heappop(scoville) * 2
        heapq.heappush(scoville, new)
        answer += 1

자료 구조 heap을 활용하여 문제를 해결할 수 있다.

profile
연세대학교 컴퓨터과학과 석사 과정

0개의 댓글