알고리즘/프로그래머스

[프로그래머스] 구명보트_JAVA

jimkwon 2022. 2. 28. 14:47
반응형

문제 링크

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

 

코딩테스트 연습 - 구명보트

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

programmers.co.kr

 

이 문제를 보고 처음 생각한 로직은 이랬다.

1. people 무게 오름차순으로 정렬

2. 가벼운 사람부터 구명보트에 limit넘기지 않도록 최대한 때려 넣기(?)

 

그런데 자꾸 테스트케이스가 오류가 나서 살펴보니...

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

최대 2명씩... 그렇다. 또 문제를 꼼꼼히 보지 못했다 ㅠㅠ

그래서 바로 로직을 바꿔 통과해버렸다.

 

1. people 무게 오름차순으로 정렬

2. 맨 뒤 무거운 사람부터 시작해서 한 명씩 보트에 태운다. 이 때, 맨 앞 가벼운 사람이 동승 가능한지 체크한다.

3. 동승이 가능하면 맨 앞 사람을 가리키는 front변수를 ++한 후, 2번 로직을 반복한다.

 

크게 살펴보면 정렬해둔 people의 양 끝을 보며 보트에 하나씩 태우고, 앞쪽 가벼운 사람과 뒷쪽 무거운 사람의 index가 만날 때까지 반복하는 방식이다.

 

// people 오름차순으로 정렬
// 맨 뒤 무거운 사람부터 시작해서 맨 앞 가벼운 사람이 동승 가능한지 체크
// 가능하면 둘 태우고 앞사람 순서 가리키는 front + 1
import java.util.Arrays;
class Solution {
    public int solution(int[] people, int limit) {
        int answer = 0;
            
        Arrays.sort(people);
        int i = people.length - 1;
        int front = 0;
        while (i >= front) {
            if (people[i] + people[front] <= limit)
                front++; //앞 사람도 태우기
            answer++; //무조건 사람 1명은 탈 수 있다.
            i--;
        }
        return answer;
    }
}

제발 문제를 꼼꼼히 읽어볼것..