Submission #1033659

#	Time	Username	Problem	Language	Result	Execution time	Memory
1033659		trucmai	Stove (JOI18_stove)	C++17	0 / 100	0 ms	348 KiB

stove

#include <bits/stdc++.h>
#include <functional>
#ifdef LOCAL
#include "/home/trcmai/code/tools.h"
#define debug(x...)                                                           \
    cerr << "\e[91m" << __func__ << ":" << __LINE__ << " [" << #x << "] = ["; \
    _print(x);                                                                \
    cerr << "\e[39m" << endl;
#else
#define debug(x...)
#endif
using namespace std;
#define all(a) a.begin(), a.end()
#define ll long long
#define endl '\n'
const int N = 1e6 + 6, LOG = 27, MOD = 1e9 + 7;
const ll INF = 1e9;
int n, k;
int T[N];
signed main()
{
    cin.tie(0)->sync_with_stdio(0);
    auto solver = [&]() {
        cin >> n >> k;
        vector<int> diff(n + 1);
        for (int i = 1; i <= n; ++i)
            cin >> T[i];
        for (int i = 1; i + 1 <= n; ++i)
            diff[i] = T[i + 1] - T[i] - 1;
        sort(diff.begin() + 1, diff.end(), greater<int>());
        int res = T[n] - T[1] + 1;
        for (int i = 1; i <= k; ++i)
            res -= diff[i];
        cout << res << endl;
    };
    int t = 1; // cin>>t;
    while (t--)
        solver();
}

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