Submission #488884

#	Time	Username	Problem	Language	Result	Execution time	Memory
488884		SlavicG	Stove (JOI18_stove)	C++17	100 / 100	21 ms	2256 KiB

stove

#include "bits/stdc++.h"
using namespace std;
 
#define ll long long
 
#define       forn(i,n)              for(int i=0;i<n;i++)
#define          all(v)              v.begin(), v.end()
#define         rall(v)              v.rbegin(),v.rend()
 
#define            pb                push_back
#define          sz(a)               (int)a.size()

void solve()
{ 
    int n, k;
    cin >> n >> k;
    --k;
    vector<int> a(n);
    for(int i = 0;i < n; ++i){
        cin >> a[i];
    }
    int ans = n;

    vector<int> v;
    for(int i = 1;i < n; ++i){
        v.pb(a[i] - (a[i - 1] + 1));
        ans += a[i] - (a[i - 1] + 1);
    }

    sort(rall(v));
    for(int i = 0;i < min(k, sz(v));++i){
        ans -= v[i];
    }

    cout << ans << "\n";
}   
 
int32_t main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    int t = 1;
    //cin >> t;
    while(t--)
    {
        solve();
    }
}

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