Submission #594392

#	Time	Username	Problem	Language	Result	Execution time	Memory
594392		zxcvbnm	Stove (JOI18_stove)	C++14	100 / 100	22 ms	2348 KiB

stove

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(nullptr);
	int n, k;
	cin >> n >> k;
	vector<int> a(n);
	for(int& i : a) {
		cin >> i;
	}
	
	priority_queue<int> q;
	q.push(0);
	ll ans = n;
	for(int i = 1; i < n; i++) {
		int len = a[i] - a[i-1] - 1;
		q.push(len);
		ans += len;
	}
	
	for(int i = 0; i < k-1; i++) {
		ans -= q.top();
		q.pop();
	}
	
	cout << ans << "\n";
}

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