Submission #545426

#	Time	Username	Problem	Language	Result	Execution time	Memory
545426		LunaMeme	Stove (JOI18_stove)	C++14	100 / 100	49 ms	1832 KiB

stove

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> ii;
typedef vector<pair<int, int>> vii;
typedef vector<int> vi;
typedef long long ll;
#define PB push_back
#define MP make_pair
#define FOR(i, x, y) for (int i = x; i < y ; i ++)

int main(){
    int n, k; cin >> n >> k;
    int prev;
    int first;
    vi gaps;
    cin >> prev;
    first = prev;
    FOR(i, 0, n - 1){
        int num; cin >> num;
        gaps.PB(num - prev - 1);
        prev = num;
    }
    sort(gaps.rbegin(), gaps.rend());
    ll ans = prev - first + 1;
    FOR(i, 0, k - 1){
        ans -= gaps[i];
    }
    cout << ans << '\n';
}

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