Submission #1009571

#	Time	Username	Problem	Language	Result	Execution time	Memory
1009571		altern23	Stove (JOI18_stove)	C++17	20 / 100	1 ms	348 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define pii pair<ll,ll>
#define fi first
#define sec second

const ll MOD = 1e9 + 7;
const ll N = 5e3 + 5;
const ll INF = 1e18;

int32_t main(){
  cin.tie(0)->sync_with_stdio(0);
  int tc = 1;
  // cin >> tc;
  while(tc--){
    ll n,k; cin >> n >> k;
    vector<ll>vec(n+5);
    for(int i=1;i<=n;i++) cin >> vec[i];
    vector<ll>seg;
    for(int i=1;i<=n;i++) seg.pb(vec[i] - vec[i-1] - 1);
    sort(seg.begin(), seg.end(), greater<ll>());
    ll ans = 0;
    for(int i=0;i<(k-1);i++) ans += seg[i];
    cout << vec[n] - vec[1] + 1 - ans << endl;
  }
}
/*
1 2 3 4 5
*/

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