Submission #421940

#	Time	Username	Problem	Language	Result	Execution time	Memory
421940		Gabp	Stove (JOI18_stove)	C++17	100 / 100	61 ms	2272 KiB

stove

#include<bits/stdc++.h>
using namespace std;
 
int main() {
  int n,k; cin >> n >> k;
  vector<int> a(n);
  for (int i = 0; i < n; i++) cin >> a[i];
  
  priority_queue<int> pq;
  for (int i = 0; i < n - 1; i++) {
    pq.push(max(0, a[i + 1] - 1 - a[i]));
  }
  
  int total = a[n - 1] - a[0] + 1;
  while (!pq.empty() && k > 1) {
    auto i = pq.top(); pq.pop();
    total -= i;
    k--;
  }
  cout << total;
}

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