Submission #76415

#	Time	Username	Problem	Language	Result	Execution time	Memory
76415		Vasiljko	Stove (JOI18_stove)	C++14	100 / 100	35 ms	11620 KiB

stove

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

typedef long long ll;
const ll MOD = 1e9+7;
const int N = 1e5+5;

int n,k;
ll t[N];
vector<ll>v;

int main()
{
    ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>t[i];
    }

    ll ans=t[n]+1-t[1];

    for(int i=2;i<=n;i++){
        v.push_back(t[i]-t[i-1]-1);
    }
    sort(v.begin(),v.end());
    reverse(v.begin(),v.end());

    k--;
    for(auto e:v){
        if(k){
            k--;
            ans-=e;
        }
    }

    cout<<ans;
    return 0;
}

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