[프로그래머스] Level2, 구명보트 (Python)

문제

https://programmers.co.kr/learn/courses/30/lessons/42885?language=python3 

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

 

코드

from collections import deque
def solution(people, limit):
    answer = 0
    people.sort()
    people = deque(people)
    while people:
        if len(people) >= 2:
            if people[0] + people[-1] <= limit:
                answer += 1
                people.popleft()
                people.pop()
            else:
                answer += 1
                people.pop()
        else:
            answer += 1
            people.pop()
    return answer

가장 큰 수 와 가장 작은 수를 더해서 limit보다 작다면 두개를 같이빼주고  limit보다 크다면 가장 큰 값만 빼준다.

 

people에 원소 갯수가 2보다 작다면 더하지 않고 바로 pop()을 해준다.

 

pop(0)을 사용하지 않고 people을 deque으로 만들어 popleft()를 사용해 시간을 단축시킨다.