Submission #1017920

#	Time	Username	Problem	Language	Result	Execution time	Memory
1017920		vjudge1	Stove (JOI18_stove)	C++17	100 / 100	19 ms	2388 KiB

stove

// J

#include "bits/stdc++.h"
using namespace std;
 
#define ar array
typedef long long ll;

//~ #define int ll

signed main() {
  ios::sync_with_stdio(0); cin.tie(0);
  
  int n, k; cin >> n >> k;
  
  
  vector<int> t(n);
  for(int i=0;i<n;i++){
    cin >> t[i];
  }
  
  vector<int> off;
  
  for(int i=0;i+1<n;i++){
    off.push_back(t[i + 1] - t[i] - 1);
  }
  
  sort(off.rbegin(), off.rend());
  k--;
  
  int ans = t[n - 1] + 1 - t[0];
  
  for(int i=0;i<k;i++){
    ans -= off[i];
  }
  
  cout<<ans<<"\n";
}

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