Submission #1331097

#	Time	Username	Problem	Language	Result	Execution time	Memory
1331097		yumemystery	Stove (JOI18_stove)	C++17	100 / 100	41 ms	1436 KiB

stove

#include <bits/stdc++.h>
using namespace std;

int main() {
    int n,k;
    cin >> n >> k;

    vector<int>guest(n);
    
    for (auto &a : guest) {
        cin >> a;
    }

    priority_queue<int,vector<int>,greater<int>>pq;

    for (int i=1; i<n; i++) {
        pq.push(guest[i]-guest[i-1]-1);
    }

    int ans = n;
    while (!pq.empty() && pq.size() > k-1) {
        ans += pq.top();
        pq.pop();
    }
    cout << ans;
}

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