Submission #1129258

#	Time	Username	Problem	Language	Result	Execution time	Memory
1129258		ziewacz	Stove (JOI18_stove)	C++20	100 / 100	16 ms	1096 KiB

stove

#include <bits/stdc++.h>
#define imie(x...) cerr << "[" #x "]: ", [](auto... $) {((cerr << $ << "; "), ...); }(x), cerr << '\n'
using namespace std;
typedef long long ll;
typedef long double ld;
const int MOD = 1000000007;


int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0);
    
    int n, k;
    cin >> n >> k;
    vector<int> T(n);
    for(int i = 0; i < n; i++) {
        cin >> T[i];
    }
    vector<int> diff(n, 0);
    for(int i = 1; i < n; i++) {
        diff[i] = T[i] - T[i - 1] - 1;
    }
    sort(diff.rbegin(), diff.rend());
    ll ans = T[n - 1] + 1 - T[0];
    for(int i = 0; i < k - 1; i++) {
        ans -= max(0, diff[i]);
    }
    cout << ans;
    return 0;
}

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