Submission #773287

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

stove

#include <bits/stdc++.h>
#define ll long long
using namespace std;

int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    int N,K; cin >> N >> K;
    vector<ll> arr(N),diff;
    for(int i = 0;i<N;++i) cin >> arr[i];
    ll ans = arr[N-1]-arr[0]+1;
    //cout << ans << "\n";
    for(int i = 0;i<N-1;++i) diff.push_back(arr[i+1]-arr[i]-1);
    sort(diff.rbegin(),diff.rend());
    for(int i = 0;i<K-1;++i) ans -= diff[i];
    cout << ans << "\n";
    return 0;
}

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