[프로그래머스-탐욕법(Greedy)] 구명보트

SE·2022년 3월 30일
0

프로그래머스

목록 보기
2/3

문제 설명

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 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:

앞에서부터 비교할 때 사용할 수 있다.

0개의 댓글

관련 채용 정보