Submission #647976

#	Time	Username	Problem	Language	Result	Execution time	Memory
647976		rohith_231	Stove (JOI18_stove)	C++17	100 / 100	28 ms	2076 KiB

stove

#include<bits/stdc++.h>
using namespace std;

int main() {
  ios::sync_with_stdio(false);
  cin.tie(NULL);
  int n,k; cin>>n>>k;
  vector<int>arr(n); for(int i=0;i<n;i++) cin>>arr[i];
  vector<int>diff(n-1);
  for(int i=0;i<n-1;i++) {
    diff[i] = arr[i+1]-arr[i]-1;
  }
   sort(diff.begin(),diff.end(),greater<int>());
  int res = arr[n-1]+1-arr[0];
  for(int i=0;i<k-1;i++) res-=diff[i];
  cout<<res<<endl;
  
  return 0;
}

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