Submission #1257422

#	Time	Username	Problem	Language	Result	Execution time	Memory
1257422		furina	Stove (JOI18_stove)	C++20	100 / 100	29 ms	1964 KiB

stove

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

int main(){
    lol N, K; cin >> N >> K;

    vector<lol> stranger(N);
    for(int i = 0; i < N; i++){
        cin >> stranger[i];
    }

    if(K >= N){ // jika pas atau overlow
        cout << N;
        return 0;
    }

    vector<lol> yaps; // gaps
    for(int i = 0; i < N; i++){
        yaps.reserve(N - 1);
    }

    for(int i = 0; i < N - 1; i++){
        lol bensin = stranger[i + 1] - stranger[i] - 1;
        yaps.push_back(bensin);
    }

    sort(yaps.begin(), yaps.end());

    lol baka = N - K; // merge
    lol panas = 0; // ekstra

    for(int i = 0; i < baka; i++){
        panas += yaps[i];
    }

    lol wakatta = N + panas;

    cout << wakatta;
}

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