[이것이 코딩 테스트다] 큰 수의 법칙

고재욱·2021년 9월 15일
0

❓ 문제 ❓

다양한 수로 이루어진 배열이 있을 때 주어진 수들을 M번 더하여 가장 큰 수를 만드는 법칙이다. 단, 배열의 특정한 인덱스에 해당하는 수가 연속해서 K번을 초과하여 더해질 수 없는 것이 법칙의 특징이다.

💯 풀이 방법 💯
전형적인 그리디 문제 풀이이다.
가장 큰 수를 K번 더하고, 두 번째 큰 수를 한 번 더하는 연산을 반복하면 된다.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main() {
	int n, m, k;
	cin >> n >> m >> k;
	vector<int> arr(n);
	for (int i = 0; i < n; i++) {
		cin >> arr[i];
	}
	sort(arr.begin(), arr.end());
	reverse(arr.begin(), arr.end());

	int answer = 0, cnt_k = 0;
	for (int i = 0; i < m; i++) {
		if (cnt_k == k) {
			cnt_k = 0;
			answer += arr[1];
			continue;
		}
		answer += arr[0];
		cnt_k++;
	}
	cout << answer;
}

0개의 댓글