[프로그래머스 고득점 kit]탐욕법 Level2 구명보트

문제 설명

 

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

 

접근 방법

이번 문제는 정확도 뿐만 아니라 효율성도 평가 사항에 포함되어 있었다. 평소 효율성에서 많은 어려움을 겪기 때문에 처음부터 문제 해결까지 사고 과정을 정리하려 한다.

 

처음에는 이중 for문을 사용하면 간단히 풀 수 있겠다는 생각을 하였다. 그리고 people 배열의 최대 크기를 보니 50,000이었기 때문에 최대 약 250,000번 반복하게 된다. 왠만하면 해결되지 않겠지만 문제 풀이의 실마리를 찾기위해 소스코드1을 작성하였고 결과는 역시 효율성 조건을 만족하지 못했다.

 

소스코드1을 작성하고 가장 크게 깨달은 점은 '배열을 이중 for문을 통해 순회하면 절대 해결할 수 없겠다' 였다. 이에 따라 해결책을 두 개 고민해보았다.

 

1.Stack과 같은 새로운 자료구조를 사용하면 어떨까?

새로운 자료구조를 사용한다 하더라도 여전히 반복문을 이중으로 돌리게되면 효율성 조건을 만족하지 못할 것 같았다.

 

2. 반복문을 하나만 써서 해결할 수 있는 방법은 없을까?

반복문을 하나만 써서 해결하기 위해서는 people 배열의 무작위로 위치해 있어서는 해결할 수 없을 것 같았다.

  2.1 배열을 오름차순(내림차순) 정렬한 상태에서는 해결할 수 있을까?

   여기까지 생각이 닿았을 때는 해결할 수 있겠다는 감이 왔다. 인덱스 변수 2개를 잡아서 하나는 앞에서 뒤로, 나머지 하나는 뒤에서 앞으로 이동하면 해결 할 수 있을 것 같아 노트에 테스트케이스를 작성하고 pseudo 코드를 작성해 보았고 이를 풀어쓰면 아래와 같다.

 

 

5

5

7

8

i는 배열의 시작을 j는 배열의 끝을 가릴킨다고 했을때, j가 가르키는 수와 i가 가르키는 수를 더했을 때 limit보다 작거나 같다면 두 사람은 보트에 탈 수 있기 때문에 i++, j-- 를 수행한다.

 

그리고 limit보다 크다면 j가 가르키는 사람 한명만 보트에 타고 j--를 수행한다.

 

그리고 i와 j가 교차되는 시점이되면 반복문을 종료한다. 

 

소스 코드1

class Solution {
    public int solution(int[] people, int limit) {
        int answer = 0;
        boolean[] ride = new boolean[people.length];
        int boatWeight = 0;
        int max = 0; int pair = 0;
        
        for(int i=0;i<people.length-1;i++){
            if(!ride[i]){
                max = 0;
                ride[i] = true;
                boatWeight = people[i];
                for(int j=i+1;j<people.length;j++){
                    if(boatWeight + people[j] <= limit && !ride[j]){
                        if(max < people[j]){
                            pair = j;
                            max = people[j];   
                        }
                    }
                }
                if(max != 0){
                    ride[pair] = true;
                    System.out.println(pair);
                }
                answer++;
            }
        }
        if(!ride[people.length - 1]){
            answer++;
            ride[people.length-1] = true;
        }
        return answer;
    }
}

소스코드2

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