프로그래머스/Lv.3

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

MINU.SHINNNN 2023. 2. 9. 20:59

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

4(n)개의 숫자로 9(s)를 만든다고 할 때, 최고의 조합은 2, 2, 2, 3 이다. s/n을 n개만큼 채워 넣고, 나머지를 분배하면 그때가 원소의 곱이 최대가 된다. 몫이 0인 경우는 만들 수 없는 경우이다.

#include <string>
#include <vector>
#include <iostream>

using namespace std;

vector<int> solution(int n, int s) {
    vector<int> answer;
    if (s%n==0){
        int k = s/n;
        for (int i=0; i<n; i++) answer.push_back(k);
    }
    else{
        int a=s/n;
        int r=s%n;
        if (a==0) return {-1};
        for (int i=0; i<n-r; i++) answer.push_back(a);
        // 나머지 분배
        for (int i=0; i<r; i++) answer.push_back(a+1);
    }
    
    return answer;
}