문제 설명
무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다. 사람들의 몸무게를 담은 배열 people과 구명보트의 무게 제한 limit가 매개변수로 주어질 때, 모든 사람을 구출하기 위해 필요한 구명보트 개수의 최솟값을 return 하도록 solution 함수를 작성해주세요.
제한 사항
- 무인도에 갇힌 사람은 1명 이상 50,000명 이하입니다.
- 각 사람의 몸무게는 40kg 이상 240kg 이하입니다.
- 구명보트의 무게 제한은 40kg 이상 240kg 이하입니다.
- 구명보트의 무게 제한은 항상 사람들의 몸무게 중 최댓값보다 크게 주어지므로 사람들을 구출할 수 없는 경우는 없습니다.
def solution(people, limit):
# (idea) 제일 작은 몸무게와 제일 큰 몸무게랑 더한다.
# 만약 그게 limit보다 크면 아무랑도 같이 탈 수 없으므로 두번째로 큰 몸무게로 넘어간다.
# 조건: 한 번에 최대 2명 탑승. 무게 제한있음. 최솟값 리턴.
c=0 # 카운팅한다.
x=0 # 앞에서부터 차례대로 순번
y=len(people)-1 # 뒤에서부터 거꾸로 차례대로 순번
people.sort() # 사람 몸무게 오름차순으로 정렬
while x<=y: # 맨 첫번째 몸무게부터 차례대로 하나씩 반복한다.
c+=1 # 카운팅 +1
if people[x]+people[y]<=limit: # 첫번째 몸무게랑 마지막 몸무게랑 더해서 limit보다 작거나 같으면,
x+=1 # 그 다음 몸무게로 넘어간다.
# 만약, limit보다 크면 무거운 몸무게 혼자 구출되게 하고, 그 다음 무거운 몸무게랑 limit 비교한다.
y-=1 # 뒤에서부터 그 다음 몸무게로 거꾸로 넘어간다.
return c # 카운팅 리턴한다.
while x<=y:
앞에서부터 비교할 때 사용할 수 있다.