ALGORITHM/PROGRAMMERS

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

0298 2021. 8. 29. 22:56

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

 

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

자연수 n 개로 이루어진 중복 집합(multi set, 편의상 이후에는 "집합"으로 통칭) 중에 다음 두 조건을 만족하는 집합을 최고의 집합이라고 합니다. 각 원소의 합이 S가 되는 수의 집합 위 조건을 만

programmers.co.kr

2021-08-29


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution {
    public int[] solution(int n, int s) {
        if(s-< 0return new int[]{-1};
        int[] answer = new int[n];
        int mod = s % n;
        for(int i = n-1; i >= 0; i--) {
            answer[i] += (s/n);
            if(mod > 0) {
                answer[i]++;
                mod--;
            }
        }
 
        return answer;
    }
}
cs

#문제풀이

1) s-n이 0보다 작은 경우에는 -1을 바로 리턴해주었다.

 

2) 그 외의 경우에는 원소간의 격차를 줄이기 위해, 주어진 s값을 일단 n등분 하였다.

 

3) 그리고 남은 나머지를 뒤에서부터 순서대로 채워넣었다.