(프로그래머스 42885) 구명보트

ps  · 2 mins read

이해

  • 구명보트는 최대 사람 2명이 탈 수 있고, 무게 제한이 있다.
  • 사람들의 몸무게가 [70kg, 50kg, 80kg, 50kg]이고 구명보트의 무게 제한이 100kg이라면2 번째 사람과 4번째 사람은 같이 탈 수 있지만 1번째 사람과 3번째 사람은 같이 탈 수 없다.
  • 사람들의 몸무게를 담은 배열 people과 구명보트의 무게 제한 limit가 주어질 때, 모든 사람을 구출하기 위해 필요한 구명보트 개수의 최솟값을 return하라.

계획

  • people 배열을 오름차순으로 정렬한다.
  • 구명보트 하나를 추가하여 제일 무게가 무거운 사람을 보트에 탑승시킨 후 아직 보트에 탑승하지 않은 사람들 중에서 제일 무게가 작은 사람이 같이 탑승할 수 있다면 같이 탑승시킨다.
  • people 배열에서 two pointer를 선언하여 첫번째 인덱스와 마지막 인덱스 값으로 초기화한 후 배열 탐색을 한다.
  • 시간 복잡도는 O(N * logN)이 된다.

실행

def solution(people, limit):
    people.sort()
    i, j = 0, len(people) - 1
    count = 0
    
    while i <= j:
        # 구명보트 추가
        count += 1
        # 마지막으로 남은 사람을 보트에 혼자 탑승시킴
        if i == j:
            break
        # j 번째 사람과 함께 탑승할 수 있는 사람이 존재(i 번째 사람)
        if people[i] + people[j] <= limit:
            # i 번째 사람 탑승
            i += 1
        # j 번째 사람 탑승
        j -= 1
            
    return count

개선

  • 2명이 탑승하는 보트의 수만 카운트해준 후 전체 사람 수에서 카운트 한 값을 빼서 전체 보트의 수를 구해준다.
def solution(people, limit):
    people.sort()
    i, j = 0, len(people) - 1
    count = 0
    
    while i < j:
        
        if people[i] + people[j] <= limit:
            i += 1
            count += 1
        j -= 1
            
    return len(people) - count

회고

  • 쉬웠고, 너무 깔끔하게 잘 푼 것 같다.