[C++/PGS] Lv.3 : 최고의 집합 (벡터, 수학)

 

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

프로그래머스

 

 

프로그래머스 레벨 3.

특별한 알고리즘은 필요하지 않고, 수학적 접근으로 해결하면 된다.

s/n (평균)에 가장 가까운 값들의 리스트가 최대 곱을 가진다.

 

힌트) 예를 들면

s = 10, n = 3일 때

s/n = 3, s%n = 1이므로

최대 곱을 가지는 리스트는 { 3, 3, 4 } 이다.

 

 

나의 풀이

#include <string>
#include <vector>
using namespace std;

vector<int> solution(int n, int s) {
    vector<int> answer;
    if(n>s) {
        answer.push_back(-1);
        return answer;
    }
    
    int avg = s / n;
    // 합을 갯수로 나눈 값과 가장 가까운 수들 -> 곱이 최대가 됨
    int last = s % n;
    // 남은 last값을 1씩 분배하면 된다!
    for(int i=0; i<n; ++i){
        if(i >= n-last) answer.push_back(avg+1);
        else answer.push_back(avg);
    }
    
    if(answer.size()<1) answer.push_back(-1);
    return answer;
}