Submission #322086

#	Time	Username	Problem	Language	Result	Execution time	Memory
322086		kwongweng	Stove (JOI18_stove)	C++14	100 / 100	52 ms	2156 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define FOR(i, a, b) for(int i = a; i < b; i++)
int main(){
	int n, k; cin >> n >> k;
	vector<int> t(n); FOR(i, 0, n) cin >> t[i];
	int ans = t[n-1] + 1 - t[0];
	vector<int> d(n-1);
	FOR(i, 0, n-1){
		d[i] = t[i+1] - t[i] - 1;
	}
	sort(d.begin(), d.end());
	FOR(i, n-k, n-1){
		ans -= d[i];
	}
	cout << ans << '\n';
	return 0;
}

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