Submission #1115596

#	Time	Username	Problem	Language	Result	Execution time	Memory
1115596		staszic_ojuz	Stove (JOI18_stove)	C++17	100 / 100	21 ms	1784 KiB

stove

// Patryk Niemczyk stov.cpp
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int main()
{
	ios_base::sync_with_stdio(0), cin.tie(0);

	int n, k;
	cin >> n >> k;

	vector<int> odstepy(n - 1);

	int ost;
	cin >> ost;
	for (int i = 1; i < n; ++i)
	{
		int akt;
		cin >> akt;
		odstepy[i - 1] = akt - ost - 1;
		ost = akt;
	}

	sort(odstepy.begin(), odstepy.end());

	auto it1 = lower_bound(odstepy.begin(), odstepy.end(), 0);
	auto it2 = upper_bound(odstepy.begin(), odstepy.end(), 0);

	int lzer = it2 - it1, wynik = n;
	int doDobrania = n - lzer - k;

	for (int i = lzer; i < lzer + doDobrania; ++i)
		wynik += odstepy[i];

	cout << wynik << '\n';
	return 0;
}

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