Submission #954297

#	Time	Username	Problem	Language	Result	Execution time	Memory
954297		hoanganh_siuuuuuuuu	Stove (JOI18_stove)	C++17	100 / 100	17 ms	4948 KiB

stove

#include <bits/stdc++.h>
#define int long long
#define f(i, a, b) for(int i = a; i <= b; i++)
#define fr(i, a, b) for(int i = a; i >= b; i--)
#define pii pair <int, int>
#define fi first
#define se second
#define pb push_back
#define eb emplace_back
#define in insert
#define vvec vector<vector<int>>
#define fast ios_base :: sync_with_stdio(0); cin.tie(0); cout.tie(0)

using namespace std;

const int N = 1e5 + 5;
const int M = 1e3 + 5;
int n, k, a[N], f[M][M];
vector <int> z;

signed main()
{
    fast;
    cin >> n >> k;
    f(i, 1, n) cin >> a[i];

    int res = a[n] + 1 - a[1];
    f(i, 1, n - 1) z.eb(a[i + 1] - a[i] - 1);
    sort(z.begin(), z.end(), greater <int> {});

    f(i, 0, k - 2) res -= z[i];

    cout << res;
}

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