[프로그래머스/Level 1] 더 맵게- python

더 맵게 

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

코딩테스트 연습 - 더 맵게

문제내용 매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다. 섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2) Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다. Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해주세요. 입력방식 - scoville의 길이는 2 이상 1,000,000 이하입니다. - K는 0 이상 1,000,000,000 이하입니다. - scoville의 원소는 각각 0 이상 1,000,000 이하입니다. 모- 든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다.

---

프로그래밍 순서

1.scoville 리스트를 heap 구조로 변경한다.

2. while 문을 scoville 리스트의 첫 번째 원소가 K보다 클 때까지 돈다.(heapq는 최소 힙 구조로 만들어지기 때문에 첫 번쨰 원소가 K보다 크면 전체가 K보다 큰 것)

  2-1 scoville 리스트에서 두 번 pop을 시켜 섞은 음식의 스코필 지수를 구한다.

  2-2 구한 섞은 음식의 스코필 지수를 scoville 리스트에 push 시킨다.

  2-3 만약 scoville 리스트에 한 개만 남았음에도 불구하여 K보다 크지 않으면 스코필 지수를 만들 수 없는 것이기 때문에 -1을 answer로 바꾼다.

3. while 문에서 빠져나오게 되면 answer를 반환한다.

---

전체코딩

 

- 코드

import heapq
 
# 섞은 음식을 스코필 지수를 만드는 함수
def makeScoville(firstS, secondS):
    return firstS + (secondS*2)
 
def solution(scoville, K):
    answer = 0
 
    # scoville heapq으로 변경
    heapq.heapify(scoville)
 
    # K이상으로 만들기
    while scoville[0] < K:
        # 스코빌 지수를 만들 수 없는 경우
        if len(scoville) == 1 and scoville[0] < K:
            answer = -1
            break
        answer += 1
        mix = makeScoville(heapq.heappop(scoville), heapq.heappop(scoville))
        heapq.heappush(scoville, mix)
    return answer

 

 

---

이전의 풀어봤던 문제를 이번에 프로그래머스 커뮤러닝에 참여하면서 다시 풀었기 때문에 프로그래머스 점수는 뜨지 않았습니다!