Submission #235473

#	Time	Username	Problem	Language	Result	Execution time	Memory
235473		ruler	Stove (JOI18_stove)	C++14	100 / 100	30 ms	2300 KiB

stove

// IOI 2021
#include <bits/stdc++.h>
using namespace std;

#define endl '\n'
#define ends ' '
#define die(x) return cout << x << endl, 0
#define all(v) v.begin(), v.end()
#define sz(x) (int)(x.size())
void debug_out() { cerr << endl; }
template <typename Head, typename... Tail>
void debug_out(Head H, Tail... T) { cerr << ends << H; debug_out(T...); }
#define debug(...) cerr << "{" << #__VA_ARGS__ << "}:", debug_out(__VA_ARGS__)
typedef long long ll;
typedef pair<int, int> pii;
const int INF = 1e9;
const ll MOD = 1e9 + 7;
 
////////////////////////////////////////////////////////////////////

const int N = 1e5 + 5;

int T[N];

int main() {

	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	mt19937 Rnd(time(0));

	int n, k, t; cin >> n >> k >> T[0];
	vector<int> V;
	for (int i = 1; i < n; i++) cin >> T[i], V.push_back(T[i] - (T[i - 1] + 1));
	int ans = T[n - 1] - T[0] + 1;
	sort(all(V)), reverse(all(V));
	for (int i = 0; i < k - 1 && i < sz(V); i++) ans -= V[i];
	cout << ans << endl;

	return 0;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:30:12: warning: unused variable 't' [-Wunused-variable]
  int n, k, t; cin >> n >> k >> T[0];
            ^

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