Submission #1012992

#	Time	Username	Problem	Language	Result	Execution time	Memory
1012992		aufan	Stove (JOI18_stove)	C++17	100 / 100	16 ms	3368 KiB

stove

#include <bits/stdc++.h>
#define int long long
#define fi first
#define se second

using namespace std;

int32_t main()
{
        ios_base::sync_with_stdio(false);
        cin.tie(NULL);
        
        int n, k;
        cin >> n >> k;

        vector<int> t(n + 1);
        for (int i = 1; i <= n; i++) cin >> t[i];

        vector<int> b;
        for (int i = 1; i <= n - 1; i++) b.push_back(t[i + 1] - t[i]);
        sort(b.rbegin(), b.rend());

        int ans = t[n] + 1 - t[1];
        for (int i = 0; i < k - 1; i++) {
                ans -= (b[i] - 1);
        }

        cout << ans << '\n';
        
        return 0;
}

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