구명보트

🔗 문제 링크

https://programmers.co.kr/learn/courses/30/lessons/42885

❔ 문제 설명

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다.

예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 50kg]이고 구명보트의 무게 제한이 100kg이라면 2번째 사람과 4번째 사람은 같이 탈 수 있지만 1번째 사람과 3번째 사람의 무게의 합은 150kg이므로 구명보트의 무게 제한을 초과하여 같이 탈 수 없습니다.

구명보트를 최대한 적게 사용하여 모든 사람을 구출하려고 합니다.

사람들의 몸무게를 담은 배열 people과 구명보트의 무게 제한 limit가 매개변수로 주어질 때, 모든 사람을 구출하기 위해 필요한 구명보트 개수의 최솟값을 return 하도록 solution 함수를 작성해주세요.

⚠️ 제한사항

• 무인도에 갇힌 사람은 1명 이상 50,000명 이하입니다.

• 각 사람의 몸무게는 40kg 이상 240kg 이하입니다.

• 구명보트의 무게 제한은 40kg 이상 240kg 이하입니다.

• 구명보트의 무게 제한은 항상 사람들의 몸무게 중 최댓값보다 크게 주어지므로 사람들을 구출할 수 없는 경우는 없습니다.

---

💡 풀이 (언어 : Java & Python)

처음 보기에는 쉬워보였는데 풀다보니 어려웠다. 나는 오름차순으로 정렬해서 작은 것부터 옆에 것과 더하는 식으로 생각했는데, 정답은 제일 작은 수와 제일 큰 수의 합을 계산하면서 체크하는 것이었다. 처음 인덱스와 마지막 인덱스를 정해주고 마지막 인덱스를 하나씩 줄여간다. 두 수를 비교해서 limit 보다 작거나 같으면 제일 작은 인덱스를 증가시켜 앞뒤로 두개를 빼주고, limit보다 크면 뒤에서만 빼줘서 하나씩 빠진다. 두 경우 모두 한번에 증가하는 카운트는 1이다. 이것을 마지막 인덱스와 처음 인덱스가 같을 때까지만 반복해준다.

Java

class Solution {
    public int solution(int[] people, int limit) {
        int answer = 0, min = 0;
        Arrays.sort(people);
        for (int max = people.length-1; max >= min; max--) {
            if (people[max] + people[min] <= limit) 
                min++;             
            answer++;
        }
        return answer;
    }
}

Python

from collections import deque

def solution(people, limit):
    answer = 0
    # 효율성 검사 때문에 deque 사용
    # deque는 sort 불가능이라 people을 sort한후 deque에 넣어줌
    people.sort()
    queue = deque(people)
    # 모든 사람이 구조될 때까지 반복
    while queue:
        # 만약 가장 큰 몸무계의 사람이 limit의 절반 이하라면
        # 그 후론 쭉 두명씩 탈 수 있음
        if queue[-1] <= limit // 2:
            # 남은 사람들이 짝수
            if len(queue) % 2 == 0:
                answer += len(queue) // 2
                break
            # 남은 사람들이 홀수
            else:
                answer += len(queue) // 2 + 1
                break
        # 가장 몸무계가 큰 사람과 가장 작은 사람의 합이 limit 이하일 때
        # 두 사람 다 리스트에서 제외 
        # 남은게 한명일 때 자기 자신과 더해지므로 예외처리
        if queue[-1] + queue[0] <= limit and len(queue) != 1:
            queue.pop()
            queue.popleft()
            answer += 1
        # 둘 다 못타면 가장 무거운 사람 혼자 구조
        else:
            queue.pop()
            answer += 1
    return answer