Submission #555046

#	Time	Username	Problem	Language	Result	Execution time	Memory
555046		colossal_pepe	Stove (JOI18_stove)	C++17	100 / 100	22 ms	2124 KiB

stove

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int n, k;
vector<int> times;
vector<int> diff;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cin >> n >> k;
    times.resize(n), diff.resize(n - 1);
    for (int &t : times) {
        cin >> t;
    }
    sort(times.begin(), times.end());
    for (int i = 0; i + 1 < n; i++) {
        diff[i] = times[i + 1] - times[i];
    }
    int ans = n;
    int k_req = n;
    sort(diff.rbegin(), diff.rend());
    while (k_req > k) {
        ans += diff.back() - 1;
        k_req--;
        diff.pop_back();
    }
    cout << ans << endl;
    return 0;
}

• Subtask 1: 20 / 20
• Subtask 2: 30 / 30
• Subtask 3: 50 / 50