코딩테스트/프로그래머스

[프로그래머스] Level 2 - 구명보트 (Python)

ye0nn 2022. 6. 24. 17:58

 

 

 

 

https://programmers.co.kr/learn/courses/30/lessons/42885

 

코딩테스트 연습 - 구명보트

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다. 예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 5

programmers.co.kr

 

 

 

 

✅ Solution

  • people을 내림차순으로 정렬한다.
  • 가장 큰 요소(people[0]) 을 보트에 추가하고, 제일 작은 요소부터 무게를 넘지 않을 때까지 추가한다.
    • len(people) >=2 일 때 반복 
  • 그리고 people에 요소가 남아있다면 보트 하나가 추가로 필요한 것이므로 answer += 1 해준다.

 

 

 

 

 ✅ Code

from collections import deque

def solution(people, limit):
    idx, answer, weight = 0, 0, 0
    people.sort(reverse=True)
    people = deque(people)

    while len(people) >= 2:
        weight = 0
        answer += 1
        weight += people.popleft()

        while people:
            if weight + people[-1] <= limit:
                weight += people.pop()
            else:
                break

    if people:
        answer += 1

    return answer