Submission #317266

#	Time	Username	Problem	Language	Result	Execution time	Memory
317266		analyticalprogrammer	Stove (JOI18_stove)	C++14	100 / 100	22 ms	2176 KiB

stove

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

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n,k;
    cin >> n >> k;
    vector<int> times(n);
    for(int& time:times) cin >> time;
    if(n==k){
      cout << n << "\n";

    }
    else if(k==1){
      cout <<times[n-1]+1-times[0] << "\n";

    }
    else{
      vector<int> diff(n-1);
      for(int i = 0;i<n-1;i++) diff[i] = times[i+1]-(times[i]+1);
      sort(diff.begin(),diff.end());
      int total = n;
      for(int i = 0;i<n-k;i++){
        total+=diff[i];
      }
      cout << total << "\n";
    }


}

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