Submission #1275954

#	Time	Username	Problem	Language	Result	Execution time	Memory
1275954		pipi	Stove (JOI18_stove)	C++20	100 / 100	14 ms	2632 KiB

stove

#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;
}

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