Submission #1014711

#	Time	Username	Problem	Language	Result	Execution time	Memory
1014711		avighna	Stove (JOI18_stove)	C++17	100 / 100	38 ms	3020 KiB

stove

#include <bits/stdc++.h>

typedef long long ll;

int main() {
    int n, k;
    std::cin >> n >> k;
    std::vector<ll> a(n);
    for (auto &i : a) {
        std::cin >> i;
    }
    std::vector<int> gaps;
    for (int i = 0; i < n - 1; ++i) {
        gaps.push_back(a[i + 1] - a[i] - 1);
    }
    std::sort(gaps.begin(), gaps.end());
    int ans = n;
    int used = n;
    int idx = 0;
    while (used > k) {
        used--;
        ans += gaps[idx++];
    }
    std::cout << ans << "\n";
}

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