자연수 n 개로 이루어진 중복 집합(multi set, 편의상 이후에는 "집합"으로 통칭) 중에 다음 두 조건을 만족하는 집합을 최고의 집합이라고 합니다.
예를 들어서 자연수 2개로 이루어진 집합 중 합이 9가 되는 집합은 다음과 같이 4개가 있습니다.
{ 1, 8 }, { 2, 7 }, { 3, 6 }, { 4, 5 }
그중 각 원소의 곱이 최대인 { 4, 5 }가 최고의 집합입니다.
집합의 원소의 개수 n과 모든 원소들의 합 s가 매개변수로 주어질 때, 최고의 집합을 return 하는 solution 함수를 완성해주세요.
최고의 집합은 s/n부터 연속인 값으로 구성될 수밖에 없다는 점을 통해 해결하면 되는 문제이다 😲
import java.util.*;
class Solution {
public int[] solution(int n, int s) {
int[] answer = new int[n];
int num = s/n;
if(num == 0) return new int[] {-1};
Arrays.fill(answer, num);
int sum = num*n; //arr 총 합
int i = n-1;
//sum이 s가 될 때까지 answer배열 값 하나씩 +1
while(sum < s) {
answer[i] += 1;
sum += 1;
i -= 1;
}
return answer;
}
}
다른 사람의 풀이를 보다 s%n, 즉 s/n으로 배열을 채우고 남은 나머지 값을 이용해 구하는 것이 더 깔끔한 것 같아 수정했다!
import java.util.*;
class Solution {
public int[] solution(int n, int s) {
int[] answer = new int[n];
int num = s/n;
if(num == 0) return new int[] {-1};
Arrays.fill(answer, num);
int sum = num*n;
int rem = s%n;
int i = n-1;
while(rem > 0) {
answer[i] += 1;
rem -= 1;
i -= 1;
}
return answer;
}
}