코딩테스트/프로그래머스
[프로그래머스] 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