[BOJ][C++] 백준 16206번: 롤케이크

https://www.acmicpc.net/problem/16206

16206번: 롤케이크

 

문제

오늘은 재현이의 생일이다. 재현이는 친구 N명에게 롤케이크를 1개씩 선물로 받았다. 롤케이크의 길이는 A1, A2, ..., AN이다. 재현이는 길이가 10인 롤케이크만 먹는다. 따라서, 롤케이크를 잘라서 길이가 10인 롤케이크를 최대한 많이 만들려고 한다.

롤케이크는 다음과 같은 과정을 통해서 자를 수 있다.

1. 자를 롤케이크를 하나 고른다. 길이가 1보다 큰 롤케이크만 자를 수 있다. 이때, 고른 롤케이크의 길이를 x라고 한다.
2. 0보다 크고, x보다 작은 자연수 y를 고른다.
3. 롤케이크를 잘라 길이가 y, x-y인 롤케이크 두 개로 만든다.

재현이는 롤케이크를 최대 M번 자를 수 있다. 이때, 만들 수 있는 길이가 10인 롤케이크 개수의 최댓값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 롤케이크의 개수 N과 자를 수 있는 최대 횟수 M이 주어진다. (1 ≤ N, M ≤ 1,000)

둘째 줄에 롤케이크의 길이 A1, A2, ..., AN이 주어진다. (1 ≤ Ai ≤ 1,000)

출력

재현이가 만들 수 있는 길이가 10인 롤케이크 개수의 최댓값을 출력한다.

 

풀이

그리디.

10의 배수가 되는 케잌들은 m번 자르면 m+1 조각이 나오므로 우선적으로 자른다

이후 그 외 케잌들은 큰 케잌부터 자른다.

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

using namespace std;

int main() {
    // 입력
    int n, m;
    cin >> n >> m;
    vector<int> v1;
    vector<int> v2;
    int x;
    for(int i=0; i<n; i++) {
        cin >> x;
        if(x%10 == 0)
            v1.push_back(x);
        else
            v2.push_back(x);
    }
    sort(v1.begin(), v1.end());
    sort(v2.begin(), v2.end(), greater<>());
    
    // 자르기
    int ans=0;
    for(int i=0; i<v1.size(); i++) {
        int tmp = v1[i]/10;
        if(tmp-1 <= m) {
            ans += tmp;
            m -= tmp-1;
        } else {
            ans += m;
            m = 0;
        }
    }
    for(int i=0; i<v2.size() && m>0; i++) {
        int tmp = v2[i]/10;
        if(tmp <= m) {
            ans += tmp;
            m -= tmp;
        } else {
            ans += m;
            m = 0;
        }
    }
    
    // 출력
    cout << ans;
    
    return 0;
}