[프로그래머스] 최고의 집합 (Java)

문제

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

코딩테스트 연습 - 최고의 집합

 

풀이

생각 보다 쉽게 풀렸던 문제 

1. 각 원소의 합이 S가 되는 수의 집합
2. 위 조건을 만족하면서 각 원소의 곱 이 최대가 되는 집합

N이 3이고 S가 9라면? 

{3, 3, 3} 이 답이 된다. 

 

N이 3이고 S가 10이라면? 

{3, 3, 4} 가 답이 된다. 

 

여러 개의 케이스를 놓고 답을 구해보니,

S를 N으로 나눈 몫에다가 나머지를 차례대로 분배한 것이 답이 되었다. 

 

N이 3이고 S가 11이라면? 

mok : 3, 나머지: 2

{3, 3 +1, 3 +1}

{3, 4, 4} 가 답이다. 

 

만약 나머지를 한쪽에 몰아넣었을 경우, 답이 되지 않는 것 확인

{3, 3, 5} = 45 

{3, 4, 4} = 48

 

이 규칙을 적용해서 제출해봤더니 통과했다.

 

소스코드

class Solution {
    public int[] solution(int n, int s) {
        // 자연수 N개로 이루어진 집합
        // 각 원소의 합이 S가 되어야 함
        int mok = s/n;
        int r = s%n;
        if(mok == 0){
            return new int[]{-1};
        }
        
        int[] answer = new int[n];
        for(int i = answer.length-1; i >= 0;i--){            
            answer[i] = mok;
            if(r > 0){
                answer[i]++;
                r--;
            }
        }

        return answer;
    }
    
}