[Algo Rhythm🕺💃] 프로그래머스 고득점 kit 저울

📚 문제 설명

하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 합니다. 이 저울의 양팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같습니다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있습니다.

 

 

저울추가 담긴 배열 weight가 매개변수로 주어질 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 return 하도록 solution 함수를 작성해주세요.

예를 들어, 무게가 각각 [3, 1, 6, 2, 7, 30, 1]인 7개의 저울추를 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21입니다.

 

제한 사항

• 저울추의 개수는 1개 이상 10,000개 이하입니다.

• 각 추의 무게는 1 이상 1,000,000 이하입니다.

 

🎯 어떻게 풀었나

첫번째 풀이(정확성 통과 o, 효율성 통과 x)

import java.util.Arrays;

class Solution {
    public int solution(int[] weight) {
        int w = 1;
        
        Arrays.sort(weight);
        
        while(true) {
            // w를 만들 수 없으면 break
            if(!isPossible(w, weight)) break;
            
            w++;
        }
        
        return w;
    }
    
    public boolean isPossible(int w, int[] weight) {
        boolean isPossible = false;
        
        for(int i = weight.length - 1; i >= 0; i--) {
            if((w - weight[i]) == 0) {
                isPossible = true;
                break;
            }
            
            if((w - weight[i]) > 0) w = w - weight[i];
        }
        
        return isPossible;
    }
}

 

최종 코드

import java.util.Arrays;

class Solution {
    public int solution(int[] weight) {
        int answer = 1;
        
        Arrays.sort(weight);
        
        for(int i = 0; i < weight.length; i++) {
            if(answer < weight[i]) break;
            
            answer = answer + weight[i];
        }
        
        return answer;
    }
}

 

🧐 왜 저렇게 풀었나

첫번째 풀이(정확성 통과 o, 효율성 통과 x)

처음에는 이 문제가 Knapsack의 하위 호환 문제라고 생각했다.

그래서 최대 w 무게까지 담을 수 있는 배낭이 있다고 가정하고 w를 1부터 하나씩 증가시켜서 배낭에 무게 w를 채울 수 있는가를 따졌다. 이 때 가장 무거운 즉, 가장 큰 숫자부터 차례로 배낭에 넣도록 코드를 작성했다.

 

최종 코드

첫번째 풀이는 정확하나 비효율적인 코드였다. 아무리 생각해도 알고리즘이 떠오르지 않아 다른 분들의 코드를 참고해서 알고리즘을 깨달았다. 이 알고리즘의 핵심은 오름차순 정렬 후 확인가능한 i번째까지의 누적 합과 i+1번째 수의 관계였다. 만약 i+1번째 수가 i번째까지의 누적 합보다 크다면 절대로 (i번째까지의 누적 합 + 1)인 수를 만들 수 없다. 그 성질을 이용해서 위와 같이 코드를 작성했다.

 

📖 무엇을 배웠나

이번 문제는 수학적 사고력을 요구하는 문제라고 생각한다. 수학적으로 사고하는 연습을 해야겠다.