Submission #1027178

#	Time	Username	Problem	Language	Result	Execution time	Memory
1027178		kustizus	Stove (JOI18_stove)	C++17	100 / 100	17 ms	3036 KiB

stove

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5;
int n,k,t[N+5];
void Solve(){
    cin>>n>>k;
    t[0]=-1;
    int ans=n,d=n;
    vector <int> v;
    for (int i=1;i<=n;++i){
        cin>>t[i];
        if (i>1) v.push_back(t[i]-t[i-1]-1);
    }
    sort(v.begin(),v.end(),greater<int>());
    while (d>k){
        --d;
        ans+=v.back();
        v.pop_back();
    }
    cout<<ans;
}
signed main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int t=1;
	// cin>>t;
	while (t--) Solve();
}

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