Submission #435258

#	Time	Username	Problem	Language	Result	Execution time	Memory
435258		ak2006	Stove (JOI18_stove)	C++14	100 / 100	26 ms	1100 KiB

stove

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using vb = vector<bool>;
using vvb = vector<vb>;
using vi = vector<int>;
using vvi = vector<vi>;
using vl = vector<ll>;
using vvl = vector<vl>;
using vc = vector<char>;
using vvc = vector<vc>;
using vs = vector<string>;
const ll mod = 1e9 + 7,inf = 1e18;
#define pb push_back
#define fast ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
void setIO()
{
    fast;
}
int main()
{
    setIO();
    int n,k;
    cin>>n>>k;
    vl durations(n - 1);
    ll prevEnd;
    cin>>prevEnd;
    prevEnd++;
    ll ans = 1;
    for (int i = 0;i<n - 1;i++){
        ll curStart;
        cin>>curStart;
        durations[i] = curStart - prevEnd;
        ans += 1 + durations[i];
        prevEnd = curStart + 1;
    }
    sort(durations.rbegin(),durations.rend());
    for (int i = 0;i<k - 1;i++)ans -= durations[i];
    cout<<ans;
    return 0;
}

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