Submission #850979

#	Time	Username	Problem	Language	Result	Execution time	Memory
850979		NamPE	Stove (JOI18_stove)	C++17	100 / 100	16 ms	3232 KiB

stove

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;

int main() {
    ios::sync_with_stdio(0);
    cout.tie(0);
    cin.tie(0);

    ll n, k;
    cin >> n >> k;

    vector<ll> a(n);

    for (ll& i : a) {
        cin >> i;
    }

    ll ans = a.back() - a[0] + 1;

    vector<ll> v;

    for(ll i = 1; i < n; i++) {
        v.push_back(a[i] - a[i - 1] - 1);
    }

    sort(v.begin(), v.end(), greater{});

    for(ll i = 0; i <= k - 2; i++) {
        ans -= v[i];
    }

    cout << ans;
}

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