Submission #1270169

#	Time	Username	Problem	Language	Result	Execution time	Memory
1270169		limits	Stove (JOI18_stove)	C++20	100 / 100	13 ms	972 KiB

stove

#pragma GCC optimize("Ofast")
#pragma GCC optimize("O3,unroll-loops")

#include <bits/stdc++.h>

using namespace std;

#define f0r(i, n) for (auto i = 0; i < (n); ++i)
#define fnr(i, n, k) for (auto i = (n); i < (k); ++i)
#define all(v) (v).begin(), (v).end()
#define pb push_back
#define F first
#define S second
#define ctn(x) cout << x << '\n'
#define forl(a, l) for (auto a : l)
#define ctl(l) for (auto &a : (l)) cout << a << ' '; cout << endl;
#define lb(v, x) (lower_bound(all(v), x) - begin(v))
#define ub(v, x) (upper_bound(all(v), x) - begin(v))
#define pq priority_queue

template <class T>
using V = vector<T>;
using ll = long long;
using vi = V<int>;
using vl = V<ll>;
using pi = pair<int, int>;
using ti = tuple<int, int, int>;
using Adj = V<vi>;
using vvi = V<vi>;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int n, k;
	cin >> n >> k;
	vi segs;
	int fst, pre = 0, cur;
	f0r(i, n) {
		cin >> cur;
		if (i) segs.pb(cur - pre - 1);
		else fst = cur;
		pre = cur;
	}
	sort(all(segs), greater<int>());
	ll sm = 0;
	f0r(i, k-1) sm += segs[i];
	ctn(cur + 1 - fst - sm);
}

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