Submission #1055440

#	Time	Username	Problem	Language	Result	Execution time	Memory
1055440		alex_2008	Stove (JOI18_stove)	C++14	100 / 100	15 ms	2548 KiB

stove

#include <iostream>
#include <vector>
#include <set>
#include <map>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
int a[N];
int main() {
	ios_base::sync_with_stdio(false); cin.tie(0);
	int n, k;
	cin >> n >> k;
	k--;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	int ans = a[n] - a[1] + 1;
	vector <int> v;
	for (int i = 1; i < n; i++) {
		v.push_back(a[i + 1] - a[i] - 1);
	}
	sort(v.begin(), v.end());
	reverse(v.begin(), v.end());
	for (int i = 0; i < k; i++) {
		ans -= v[i];
	}
	cout << ans << "\n";
}

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