이해
- 구명보트는 최대 사람 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
회고
- 쉬웠고, 너무 깔끔하게 잘 푼 것 같다.