Submission #1257221

#	Time	Username	Problem	Language	Result	Execution time	Memory
1257221		rpthirster	Stove (JOI18_stove)	C++20	100 / 100	16 ms	1988 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define fio() ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)

#define lo long long
#define fi first
#define se second

lo n,k;

lo guests[100005];
vector<lo> diff;



int main()
{   
    fio();

    cin >> n >> k;

    for(int i=1; i<=n; i++){
        cin >> guests[i];
        if(i>=2){
            diff.push_back(guests[i]- guests[i-1] -1);
        }
    }

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


    lo ans = guests[n] - guests[1] +1;

    for(int i=0; i<k-1; i++){
        ans -= diff[i];
    }

    cout << ans;


    
    return 0; 
}

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