본문 바로가기

알고리즘

프로그래머스 - 더 맵게(Python)

반응형

문제 설명

알고리즘 고민

스코빌 지수 배열을 힙 배열로 변환한 후 앞의 두 요소를 heappop으로 빼온 후 계산하여 다시 scoville에 넣어준다. 이때 scoville 지수 배열의 길이가 2보다 작은 경우는 결국 모든 음식의 스코빌 지수를 K 이상으로 만들지 못한 것이므로 -1을 반환하도록 하여 해결하였다. 문제의 어떤 상황에서 종료가 안되는지 되는지 잘 판단해야겠다.

코드

반응형