Submission #94474

#	Time	Username	Problem	Language	Result	Execution time	Memory
94474		fedoseevtimofey	Stove (JOI18_stove)	C++14	100 / 100	22 ms	1912 KiB

stove

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0); cout.setf(ios::fixed); cout.precision(20);
    #ifdef LOCAL
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
    #endif
    int n, k;
    cin >> n >> k;
    int lst = -1;
    vector <int> dif;
    for (int i = 0; i < n; ++i) {
        int t;
        cin >> t;
        if (lst != -1) {
            dif.push_back(t - lst);
        }
        lst = t + 1;
    }
    int cnt = n;
    int ans = n;
    sort(dif.begin(), dif.end());
    for (auto x : dif) {
        if (cnt > k) {
            --cnt;
            ans += x;
        }
    }
    cout << ans << '\n';
}

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