Submission #1318370

#	Time	Username	Problem	Language	Result	Execution time	Memory
1318370		eshanahmed	Stove (JOI18_stove)	C++20	100 / 100	36 ms	1552 KiB

stove

// ﷽
#include <bits/stdc++.h>
using namespace std;
#define ll long long
void solve(){
    ll n,k;
    cin>>n>>k;
    vector<ll> s;
    k--;

    ll first, prev;
    cin >> first;
    prev = first;

    for(int i=1;i<n;i++){
        ll x;
        cin >> x;
        s.push_back(x - prev - 1);
        prev = x;
    }

    ll ans = (prev + 1) - first;

    sort(s.rbegin(), s.rend()); 

    for(int i=0;i<k && i<(int)s.size();i++){
        ans -= s[i]; 
    }

    cout << ans << endl;
}
int main(){
   ll n=1;
   while(n--){
     solve();
   }
}

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