Submission #1257653

#	Time	Username	Problem	Language	Result	Execution time	Memory
1257653		damasen	Stove (JOI18_stove)	C++20	100 / 100	29 ms	1864 KiB

stove

#include <bits/stdc++.h>

using namespace std;

int main(){
    int n, k;
    cin >> n >> k;
    vector<long long> guests (n);
    for(int i = 0; i < n; i++){
        cin >> guests[i];
    }
    vector<long long> gaps(n-1);
    for(int i = 0; i < n-1; i++){
        gaps[i] = guests[i+1]-(guests[i]+1);
    }
    sort(gaps.rbegin(), gaps.rend());
    long long ans = n;
    
    for(int i = k-1; i < n-1 && gaps[i] > 0; i++){
        ans += gaps[i];
    }
    cout << ans;
}

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