Submission #994550

#	Time	Username	Problem	Language	Result	Execution time	Memory
994550		APROHACK	Stove (JOI18_stove)	C++17	100 / 100	16 ms	2432 KiB

stove

#include <bits/stdc++.h>
#define ll long long
#define ff first
#define ss second
#define pb push_back
using namespace std;
ll n, k;
vector<int>t;
vector<int>restas;

void tc(){
    cin >> n >> k;

    for(int i = 0 ; i < n ; i ++){
        int temp;
        cin >> temp;
        if(i > 0)restas.pb(temp - t.back() - 1);
        t.pb(temp);
    }
    sort(restas.begin(), restas.end());
    int total = t.back() + 1 - t[0];
    for(int i = 0 ; i < k - 1 ; i ++){
        if(!restas.empty()){
            total -= restas.back();
            restas.pop_back();
        }
    }
    cout << total << "\n";
}

void end_tc(){

}

int main() {
    cin.tie(0);
    cout.tie(0);

    ios::sync_with_stdio(NULL);
    int t = 1;
    while(t--){
        tc();
        end_tc();
    }




}

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