Submission #1275954

#TimeUsernameProblemLanguageResultExecution timeMemory
1275954pipiStove (JOI18_stove)C++20
100 / 100
14 ms2632 KiB
#include <bits/stdc++.h>
using namespace std;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int N, K;
    cin >> N >> K;
    vector<long long> T(N);
    for (int i = 0; i < N; i++) cin >> T[i];

    // Nếu chỉ có 1 khách hoặc K >= N thì chỉ cần bật riêng từng khách
    if (K >= N) {
        cout << N << "\n";
        return 0;
    }

    vector<long long> gaps;
    for (int i = 0; i < N - 1; i++) {
        long long g = T[i + 1] - (T[i] + 1);
        gaps.push_back(g);
    }

    sort(gaps.begin(), gaps.end());

    long long total_gap = 0;
    for (long long g : gaps) total_gap += g;

    // loại bỏ (K-1) khoảng trống lớn nhất
    long long save = 0;
    for (int i = 0; i < K - 1; i++) {
        save += gaps[gaps.size() - 1 - i];
    }

    long long ans = N + total_gap - save;
    cout << ans << "\n";

    return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...