Submission #885421

#	Time	Username	Problem	Language	Result	Execution time	Memory
885421		raspy	Stove (JOI18_stove)	C++14	100 / 100	42 ms	3048 KiB

stove

#include <iostream>
#include <queue>
#include <vector>

#define int long long

using namespace std;

int a[200005];

int32_t main()
{
	int n, k;
	cin >> n >> k;
	priority_queue<int, vector<int>, greater<int>> pq;
	int tr = 1;
	for (int i = 0; i < n; i++)
	{
		cin >> a[i];
		if (i && a[i] != a[i-1]+1)
		{
			tr++;
			pq.push(a[i]-a[i-1]-1);
		}
	}
	int rez = n;
	while (tr > k)
	{
		rez += pq.top();
		pq.pop();
		tr--;
	}
	cout << rez << "\n";
	return 0;
}

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