구명보트 - 프로그래머스 - 파이썬 풀이

문제 풀이

1. people을 가벼운 순으로 sort 합니다.
2. 이후 제일 무거운 사람을 태운 후, 제일 가벼운 사람을 태울 수 있으면 태웁니다.
3. 모든 사람들이 탈 때까지 2번의 로직을 반복합니다.
4. list 자료형의 경우 0번째 index pop의 비용이 O(n)이므로 deque자료형을 이용합니다.

 

클린 코드 작성법

1. 문제 이해를 위한 적절한 변수명을 짓습니다.
   capacity, heaviest_person, lightest_person

2. if capacity_left_for_lightest_person(people, capacity) 처럼 함수을 이용해 가독성을 높일 수 있씁니다.
   if 다음에 boolean 값을 return 하는 함수를 작성합니다.

3. 값의 비교시 작은 값은 왼쪽, 큰 값은 오른쪽에 놓습니다.
   lightest_person <= capacity

 

해답 코드

from collections import deque

def solution(people, limit):
    people.sort()
    people = deque(people)
    answer = 0
    while people:
        capacity = limit
        heaviest_person = people.pop()
        capacity -= heaviest_person
        if capacity_left_for_lightest_person(people, capacity):
            people.popleft()
        answer+=1
    return answer

def capacity_left_for_lightest_person(people, capacity):
    if not people:
        return False
    lightest_person = people[0]
    if lightest_person <= capacity:
        return True

 

도움이 됐다면 공감버튼을 눌러주세요! 질문이 있다면 댓글 달아주세요!

 

문제 링크

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