190115 탐욕법 (6) 저울
탐욕법 (6) 저울
문제 설명
하나의 양팔 저울을 이용하여 물건의 무게를 측정하려고 합니다. 이 저울의 양팔의 끝에는 물건이나 추를 올려놓는 접시가 달려 있고, 양팔의 길이는 같습니다. 또한, 저울의 한쪽에는 저울추들만 놓을 수 있고, 다른 쪽에는 무게를 측정하려는 물건만 올려놓을 수 있습니다.
저울추가 담긴 배열 weight가 매개변수로 주어질 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값을 return 하도록 solution 함수를 작성해주세요.
예를 들어, 무게가 각각 [3, 1, 6, 2, 7, 30, 1]인 7개의 저울추를 주어졌을 때, 이 추들로 측정할 수 없는 양의 정수 무게 중 최솟값은 21입니다.
제한 사항
- 저울추의 개수는 1개 이상 10,000개 이하입니다.
- 각 추의 무게는 1 이상 1,000,000 이하입니다.
입출력 예
weight | return |
---|---|
[3, 1, 6, 2, 7, 30, 1] | 21 |
풀이 과정
- 배열을 오름차순 정렬하고, (현재 원소까지의 누적합+1)이 다음 원소보다 작은 경우 (현재 원소까지의 누적합+1)을 반환
import java.util.*;
class Solution {
public int solution(int[] weight) {
int answer = 0;
Arrays.sort(weight);
for(int i=0;i<weight.length;i++){
int tmp=weight[i];
if(answer+1<tmp){
return answer+1;
}else{
answer+=tmp;
}
}
return answer+1;
}
}