Submission #1017925

#	Time	Username	Problem	Language	Result	Execution time	Memory
1017925		vjudge1	Stove (JOI18_stove)	C++17	100 / 100	16 ms	1504 KiB

stove

#include "bits/stdc++.h"
using namespace std;
#define ar array
typedef long long 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