Submission #903936

#	Time	Username	Problem	Language	Result	Execution time	Memory
903936		zhasyn	Stove (JOI18_stove)	C++17	100 / 100	20 ms	2524 KiB

stove

#include <bits/stdc++.h>
using namespace std;
int arr[100010];
int main() {
	ios_base::sync_with_stdio(false);
    cin.tie(NULL);
  	int n, k, ans = 0;
  	cin >> n >> k;
  	for(int i = 0; i < n; i++){
  		cin >> arr[i];
  	}
  	sort(arr, arr + n);
  	vector <int> w;
  	int cur = arr[0], fir = arr[0], cnt = 1;
  	for(int i = 1; i < n; i++){
  		if(arr[i] - cur <= 1) cur = arr[i];
  		else{
  			ans += (cur - fir + 1);
  			cnt++;
  			w.push_back({arr[i] - cur - 1});
  			cur = fir = arr[i];
  		}
  	}
  	ans += (cur - fir + 1);
  	sort(w.begin(), w.end());
  	for(int i = 0; i < (cnt - k); i++){
  		ans += w[i];
  	}
  	cout << ans;
  	return 0;
}

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