Submission #548875

#	Time	Username	Problem	Language	Result	Execution time	Memory
548875		DemonLord154	Stove (JOI18_stove)	C++14	100 / 100	59 ms	1400 KiB

stove

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
 
int solve(int n, vector<int> & times, int k){
 	vector<int> gaps(n-1);
  for (int i=0;i<n-1;i++){
    gaps[i]=times[i+1]-times[i]-1;
  }
  sort(gaps.begin(),gaps.end(),greater<int>());
  int answer= times[n-1]-times[0]+1;
  for (int i=0;i<k-1;i++){
    answer-=gaps[i];
  }
  return answer;
}
int main(){
  int n,k;
  cin>>n>>k;
  vector <int> times(n);
  for (int i=0;i<n;i++){
    cin>>times[i];
  }
  cout<<solve(n,times,k)<<"\n";
  return 0;
}

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