프로그래머스/Lv. 2

[프로그래머스 코딩테스트] 구명보트(Java)

Sigfriede 2023. 6. 7. 01:00

  문제 설명

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

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

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

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

 

  제한사항

  • 무인도에 갇힌 사람은 1명 이상 50,000명 이하입니다.
  • 각 사람의 몸무게는 40kg 이상 240kg 이하입니다.
  • 구명보트의 무게 제한은 40kg 이상 240kg 이하입니다.
  • 구명보트의 무게 제한은 항상 사람들의 몸무게 중 최댓값보다 크게 주어지므로 사람들을 구출할 수 없는 경우는 없습니다.

 

  입출력 예

people limit return
[70, 50, 80, 50] 100 3
[70, 80, 50] 100 3
import java.util.Arrays;
class Solution {
    public int solution(int[] people, int limit) {
        int answer = 0;
        Arrays.sort(people);
        int i = 0;
        int j = people.length - 1;
        while (i <= j) {
            if (people[i] + people[j] <= limit) {
                answer++;
                i++;
                j--;
            } else {
                answer++;
                j--;
            }
        }
        return answer;
    }
}

  Arrays 클래스의 sort를 이용하여 people을 오름차순으로 정렬합니다.

  변수 i와 j를 생성합니다. 투포인터 알고리즘을 이용하여 풀이할 것입니다. 이 코드에서는 포인터를 배열의 처음과 끝으로 할당하여 포인터가 각각 반대 방향으로 진행할 것입니다.

  while문이 i가 j보다 크거나 같아질 때까지 반복합니다.

  if문에서 people의 i번째 원소와 people의 j번째 원소를 더한 값이 limit보다 큰지 확인합니다. 조건을 충족한다면 answer가 증가합니다. i와 j는 반대 방향으로 진행해야 하기 때문에 i는 증가하고 j는 감소합니다.

  만약 앞선 조건을 충족하지 않는다면, 즉 두 수의 합이 limit보다 크다면 몸무게가 많이 나가는 사람을 태운다는 가정 하에 answer가 증가하고  j가 감소합니다.