https://programmers.co.kr/learn/courses/30/lessons/42626
코딩테스트 연습 - 더 맵게 | 프로그래머스
매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다. 섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2) Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다. Leo가 가진
programmers.co.kr
정렬하고 최소값을 찾을때는 최소힙으로 풀어야 시간복잡도 낮출 수 있다
import heapq
def solution(scoville, K):
answer = 0
heapq.heapify(scoville)
while True:
min1 = heapq.heappop(scoville)
if min1 >= K:
break
elif len(scoville) == 0:
answer = -1
break
min2 = heapq.heappop(scoville)
_sum = min1 + 2*min2
heapq.heappush(scoville, _sum)
answer += 1
return answer
'알고리즘 > 프로그래머스(Python)' 카테고리의 다른 글
[알고리즘] 프로그래머스 여행경로 / python (0) | 2020.02.22 |
---|---|
[알고리즘] 프로그래머스 N으로 표현 / python (0) | 2020.02.20 |
[알고리즘] 프로그래머스 큰 수 만들기 / python (0) | 2020.02.16 |
[알고리즘] 프로그래머스 가장 큰 수 / python (0) | 2020.02.15 |
[알고리즘] 프로그래머스 체육복 / python (0) | 2020.02.15 |