Submission #1308803

#	Time	Username	Problem	Language	Result	Execution time	Memory
1308803		liptonek	Stove (JOI18_stove)	C++20	100 / 100	14 ms	2864 KiB

stove

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

int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    int n,k;
    cin>>n>>k;

    vector<long long> t(n);

    for(int i=0; i<n; i++)
    {
        cin>>t[i];
    }

    vector<long long> gaps;

    if(n>1)
    {
        gaps.reserve(n-1);

        for(int i=0; i<n-1; i++)
        {
            long long len=t[i+1]-t[i]-1;

            gaps.push_back(len);
        }
    }

    long long total=(t[n-1]+1)-t[0];

    sort(gaps.begin(), gaps.end(), greater<long long>());

    int cuts=k-1;

    for(int i=0; i<gaps.size(); i++)
    {
        if(i<cuts)
        {
            total-=gaps[i];
        }
        else
        {
            break;
        }
    }

    cout<<total<<endl;

    return 0;
}

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