Submission #903932

#	Time	Username	Problem	Language	Result	Execution time	Memory
903932		zhasyn	Stove (JOI18_stove)	C++17	0 / 100	1028 ms	352 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);
  	set <int> st;
  	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++;
  			st.insert({arr[i] - cur - 1});
  			cur = fir = arr[i];
  		}
  	}
  	ans += (cur - fir + 1);
  	while(cnt > k){
  		ans += *st.begin();
  		st.erase(st.begin());
  		cnt--;
  	}
  	cout << ans;
  	return 0;
}

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