제출 #1033660

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1033660		trucmai	Stove (JOI18_stove)	C++17	100 / 100	19 ms	2908 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 - 1; ++i)
            res -= diff[i];
        cout << res << endl;
    };
    int t = 1; // cin>>t;
    while (t--)
        solver();
}

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