Submission #718322

#	Time	Username	Problem	Language	Result	Execution time	Memory
718322		mshandilya	Stove (JOI18_stove)	C++17	100 / 100	18 ms	2264 KiB

stove

//required macros and template
#include <bits/stdc++.h>
using namespace std;

//type definitions
#define vc vector
typedef vc<int> vi;

//main method
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, k, ans;
    cin>>n>>k;
    ans = n;
    vi t(n), x;
    for(int i = 0; i<n; i++) {
        cin >> t[i];
        if(i and t[i]!=t[i-1]+1)
            x.push_back(t[i]-t[i-1]-1);
    }
    sort(x.begin(), x.end());
    for(int i = x.size()-k; i>=0; i--)
        ans += x[i];
    cout<<ans<<"\n";
    return 0;
}

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