[level 3] 최고의 집합 - 12938

문제 링크

문제 설명

자연수 개로 이루어진 중복 집합 중에 다음 두 조건을 만족하는 집합을 최고의 집합이라고 합니다.

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

집합의 원소의 개수 과 모든 원소들의 합 가 매개변수로 주어질 때, 최고의 집합을 return 하는 solution 함수를 완성해주세요.

제한 사항

• 최고의 집합은 오름차순으로 정렬된 1차원 배열로 return 해주세요.
• 만약 최고의 집합이 존재하지 않는 경우에 크기가 1인 1차원 배열에 -1을 채워 return 해주세요.
• 은 1 이상 10,000 이하인 자연수입니다.
• 은 1 이상 100,000,000 이하인 자연수입니다.

---

입출력 예시

n	s	result
2	9	[4, 5]
2	1	[-1]
2	8	[4, 4]

입출력 예 설명

예시 1: 인 경우, 합이 9가 되는 집합은 8, 7, 6, 5 등이 있습니다. 이 중 곱이 가장 큰 5가 최고의 집합입니다.

---

구현 코드

import java.util.*;

class Solution {
  public int[] solution(int n, int s) {
        if(n > s) {
            return new int[] {-1};
        }
        int[] answer = new int[n];
        if(s % n ==0) {        //나눠지는 경우
            for(int i = 0 ; i < n; i++) {
                answer[i] = s / n;
            }

        }else {                //안나눠지는 경우
            int rem = s % n;
            int po = n - rem;    //1씩 더해줘야하는 인덱스 위치
            for(int i = 0; i < po; i++) {  //po까지는 원래대로 추가
                answer[i] = s / n;
            }
            for(int i = po; i < n; i++) {  //po부터 n까지는 1씩 추가
                answer[i] = s / n +1;
            }
        }
        return answer;
    }

}

성능 요약

메모리: 53.6 MB, 시간: 0.28 ms

채점결과

정확성: 70.0

효율성: 30.0

합계: 100.0 / 100.0