Submission #1004886

#	Time	Username	Problem	Language	Result	Execution time	Memory
1004886		MarwenElarbi	Stove (JOI18_stove)	C++17	100 / 100	40 ms	2408 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#define fi first
#define se second
#define ll long long
#define pb push_back
#define ii pair<int,int>
const int nax=5e2+5;
int main()
{
    
    int n,k;
    cin>>n>>k;
    int tab[n];
    for (int i = 0; i < n; ++i)
    {
        cin>>tab[i];
    }
    vector<int> nabba;
    for (int i = 1; i < n; ++i)
    {
        nabba.pb(tab[i]-tab[i-1]-1);
    }
    sort(nabba.begin(),nabba.end());
    int sum=tab[n-1]+1-tab[0];
    for (int i = nabba.size()-1; i >= (int)nabba.size()-k+1  ; --i)
    {
        sum-=nabba[i];
    }
    cout <<sum<<endl;
}

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