Submission #1283641

#	Time	Username	Problem	Language	Result	Execution time	Memory
1283641		lunarecho	Stove (JOI18_stove)	C++20	100 / 100	15 ms	2376 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define ll long long

int main() 
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, k;
    cin>>n>>k;
    vector<ll> a(n);
    for(auto &it : a)
        cin>>it;
    if(k >= n)
    {
        cout<<n<<'\n';
        return 0;
    }
    vector<ll> d;
    for(int i=0;i<n-1;++i)
    {
        d.push_back(a[i + 1] - (a[i] + 1));
    }
    int ans = n;
    sort(d.begin(),d.end());
    for(int i=0;i<n-k;++i)
        ans += d[i];
    cout<<ans<<'\n';

    return 0;
}

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