문제

자연수 N개로 이루어진 집합 중에, 각 원소의 합이 S가 되는 수의 집합은 여러 가지가 존재합니다.
최고의 집합은, 위의 조건을 만족하는 집합 중 각 원소의 곱이 최대가 되는 집합을 의미합니다.
집합 원소의 개수 n과 원소들의 합 s가 주어지면, 최고의 집합을 찾아 원소를 오름차순으로 반환해주는 bestSet 함수를 만들어 보세요.
만약 조건을 만족하는 집합이 없을 때는 배열 맨 앞에 –1을 담아 반환하면 됩니다.
예를 들어 n=3, s=13이면 [4,4,5]가 반환됩니다.
(자바는 집합이 없는 경우 크기가 1인 배열에 -1을 담아 반환해주세요.)

풀이 1

import java.util.Arrays; //테스트로 출력해 보기 위한 코드입니다.

public class BestSet {
  public int[] bestSet(int n, int s){
    int[] answer = new int[n];

    if(s<n) return new int[]{-1};

    for(int i=0; i<n; i++) {
      answer[i] = (s/(n-i));
      s = s - (s/(n-i));    	
    }  

    Arrays.sort(answer);

    return answer;
  }
  
  public static void main(String[] args) {
    BestSet c = new BestSet();
    //아래는 테스트로 출력해 보기 위한 코드입니다.
    System.out.println(Arrays.toString(c.bestSet(3,13)));
  }
}

얼떨결에 풀었다. 왠지 s와 n을 나누어 그 수를 배열에 넣어주고 다시 s에 빼주고 같은 과정을 반복하면 답인 것 같아서 풀었더니 맞았다.

풀이 2

import java.util.Arrays;

public class BestSet {

    public int[] bestSet(int n, int s){
        int[] answer = null; 
        if(n>s) {
            answer = new int[1];
            answer[0] = -1;
        } else {
            answer = new int[n];
            int i = 0;
            while(s>0) {
                answer[(i%n)]++;
                i++;
                s--;
            }
        }
        
        Arrays.sort(answer);
        
        return answer;  
    }
    public static void main(String[] args) {
        BestSet c = new BestSet();
        //아래는 테스트로 출력해 보기 위한 코드입니다.
        System.out.println(c.bestSet(3,13));
    }

}