9-wnsmir

[wnsmir]

🔗 문제 링크

이것이 취업을 위한 코딩테스트이다 책에서 발췌한 문제라 링크는 따로 없습니다.

예를들어 2 4 5 4 6으로 이루어진 배열이 있을때 M이 8이고 K가 3이라면 K번만큼 같은수를 연속해서 더할 수 있고 총 더할 수 있는 수는 M번 즉 8번이다. M번 더했을때 가장 큰 수를 구하라.

입력은 첫째줄에 N, M, K 가 주어지고 N은 배열의 크기, M은 총 더하는 수 K는 같은수가 연속으로 더해지는 수 이다.
둘째줄에 N개의 수가 공백으로 구분되어 주어진다.

입력 예시
5 8 3
2 4 5 4 6

출력 예시 46

✔️ 소요된 시간

20min

✨ 수도 코드

대표적인 쉬운 그리디 문제로, 연속해서 올 수 있는 수의 제한이 있다는건 그다음 수는 두번쨰로 큰수를 더해주고 다시 연속해서 가장 큰수를 더해주면 된다. 즉 내가 필요한 숫자는 가장큰수와 두번째로 큰수 두개만 필요한 것이다.

while문을 크게 돌며 M번을 채우도록 했고,
for문을 돌며 k번 가장 큰수를 더해주며

for문을 나오면 M번을 넘지 않는 선에서 두번째로 큰수를 더해주는 식으로 코드를 구성하였습니다.

[kangrae-jo]

처음 문제를 읽고나서는,
연속해서 K번 사용하면 그 수는 버려야하는 줄 알았네용

문제를 다시 읽어보니까 준용님 설명대로 가장 큰 값, 두번째로 큰 값만 저장하면 되고
가장 큰 값을 K번 더하고, 두번째로 큰 값을 1번 더하고, 이러면 되겠더라구요

CPP CODE

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

int main() {
    int N, M, K;
    cin >> N >> M >> K;

    vector<int> input(N);
    for (int i = 0; i < N; i++) {
        cin >> input[i];
    }
    sort(input.begin(), input.end());
    int first = input[N - 1];
    int second = input[N - 2];

    int result = 0;
    for (int i = 0; i < M;) {
        for (int j = 0; j < K && i < M; j++, i++) {
            result += first;
        }
        if (i >= M) break;
        
        result += second;
        i++;
    }
    cout << result;

    return 0;
}

그리고 이거를 이렇게 for문으로 돌리지 말고 수학적으로 미리 계산하면
O(1) 시간복잡도를 가지는 코드도 완성할 수 있을 듯 하네요!

아래 코드는 처음 푼 방식에 대한 코드입니돠.
푼게 아까워서... 끼워팔기 하겠습니다.

CPP CODE

#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

int main() {
    int N, M, K;
    cin >> N >> M >> K;

    vector<int> input(N);
    for (int i = 0; i < N; i++) {
        cin >> input[i];
    }
    sort(input.begin(), input.end());

    int result = 0;
    int index = N-1;
    for (int i = 0; i < M;) {
        for (int j = 0; j < K; j++) {
            result += input[index];
            i++;
            if (i >= M) break;
        }
        index--;
    }
    cout << result;

    return 0;
}

[g0rnn]

숫자를 두개 밖에 사용하지 않는다는 점과 k번의 가장 큰수 + 1번의 두번째 큰 수의 합의 연속이기 때문에 수학적으로 풀어보았습니다. 입력을 받을 때 가장 큰수와 두번째로 큰 수만 받았어요

    int part = fir * k + sec;
    if(m <= k) cout<<fir*k;
    else {
        int cnt = m / (k+1);
        int rest = m % (k+1);
        cout << part * cnt + fir * rest;
    }