Submission #1196547

#	Time	Username	Problem	Language	Result	Execution time	Memory
1196547		amanthaband	Stove (JOI18_stove)	C++20	100 / 100	31 ms	1896 KiB

stove

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

using namespace std;

int main() {
    int n, k;
    cin >> n >> k;
    vector<int> a(n);
    for (int &x : a) cin >> x;
    sort(a.begin(), a.end());
    
    vector<int> d;
    for (int i = 1; i < n; i++) d.push_back(a[i] - a[i - 1] - 1);
    
    sort(d.begin(), d.end(), greater<int>());
    
    k = n - k;
    while (k-- && !d.empty()) {
        n += d.back();
        d.pop_back();
    }
    
    cout << n << endl;
    return 0;
}

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