Submission #516469

#	Time	Username	Problem	Language	Result	Execution time	Memory
516469		srivatsav_kannan	Stove (JOI18_stove)	C++14	100 / 100	26 ms	2360 KiB

stove

#include <iostream>
#include <fstream>
#include <vector>
#include <set>
#include <queue>
#include <cmath>
#include <map>
#include <algorithm>
#include <numeric>
#include <stack>
#include <cstring>
#include <bitset>
using namespace std;
#define int long long int
#define mod 1000000007
#define endl '\n'
signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n,k; cin >> n >> k;
    vector<int> ans;
    int prev; cin >> prev;
    for (int i = 1; i < n; i++){
        int a; cin >> a;
        ans.push_back(a-prev);
        prev = a;
    }
    int anss = 0;
    sort(ans.begin(), ans.end());
    for (int i = 0; i < n-k; i++){
        ans[i]--;
        anss += ans[i];
    }
    cout << anss+n << endl;
    return 0;
}

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