[프로그래머스] 구명보트

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

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

 

최대 2명씩밖에 탈 수가 없으므로 가장 무거운 사람을 기준으로 2명을 태워 보낼 수 있는지 없는지를 판단한다.

오름차순으로 정렬된 people 에서 end -= 1 을 해주면서 2명을 태워 보낼 수 있으면 start += 1 시켜준다. start 가 end 를 넘어갈 때 answer 을 반환해주면 된다.

 

def solution(people, limit):
    answer = 0
    people.sort()
    start, end = 0, len(people) - 1
    
    while start <= end:
        answer += 1
        
        if people[start] + people[end] <= limit:
            start += 1
        
        end -= 1

    return answer

print(solution([70, 50, 80, 50], 100)) # 3