Submission #1033620

#	Time	Username	Problem	Language	Result	Execution time	Memory
1033620		trucmai	Stove (JOI18_stove)	C++17	0 / 100	0 ms	344 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;
        for (int i = 1; i <= n; ++i) {
            cin >> T[i];
            if (i > 1)
                diff.emplace_back(T[i] - T[i - 1] - 1);
        }
        if (k == 1) {
            int sum = 0;
            for (int i = 1; i <= n; ++i)
                sum += T[i];
            cout << sum;
            return;
        }
        --k;
        sort(diff.begin(), diff.end(), greater<int>());
        int res = k + 1;
        for (int i = k; i < diff.size(); ++i)
            res += diff[i];
        cout << res << endl;
    };
    int t = 1; // cin>>t;
    while (t--)
        solver();
}

Compilation message (stderr)

stove.cpp: In lambda function:
stove.cpp:41:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   41 |         for (int i = k; i < diff.size(); ++i)
      |                         ~~^~~~~~~~~~~~~

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