Submission #391507

#	Time	Username	Problem	Language	Result	Execution time	Memory
391507		giorgikob	Stove (JOI18_stove)	C++14	100 / 100	25 ms	2380 KiB

stove

#include<bits/stdc++.h>
#define ll long long
#define ff first
#define ss second
#define pb push_back
using namespace std;

const int N = 4e5+5, mod = 3;

int n,k;
vector<int>v;
int A[N];
ll answer;

inline void test_case(){

    cin >> n >> k;
    for(int i = 1; i <= n; i++){
        cin >> A[i];
    }

    for(int i = 1; i < n; i++){
        v.pb(A[i+1]-A[i]-1);
    }

    sort(v.begin(),v.end());

    for(int i = 0; i < n-k; i++){
        answer += v[i];
    }

    cout << answer+n << endl;
}

 main(){
    ios::sync_with_stdio(0);

    int T = 1;
    //cin >> T;
    while(T--){
        test_case();
    }
}

Compilation message (stderr)

stove.cpp:35:7: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   35 |  main(){
      |       ^

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