Submission #146080

#	Time	Username	Problem	Language	Result	Execution time	Memory
146080		MathStudent2002	Stove (JOI18_stove)	C++14	100 / 100	128 ms	6052 KiB

stove

//wait darn

#include<bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef set<ll>::iterator slt;

ll st,fi;
multiset<ll> df;
ll K;

void input() {
    int N; cin >> N >> K;

    ll pre, cur;
    for(int i = 0; i < N; i++) {
        cin >> cur;
        if(i == 0) st = cur;
        else df.insert(pre-cur);

        if(i == N-1) fi = cur+1;
        else pre = cur+1;
    }
}

void solve() {
    K--;
    ll ans = fi-st;
    int i = 0;
    slt it = df.begin();
    for(; i < K; i++, it++) {ans += (*it);}
    cout << ans << endl;
}

int main() {
    input();
    solve();
}

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