Submission #901281

#	Time	Username	Problem	Language	Result	Execution time	Memory
901281		KienTran	Stove (JOI18_stove)	C++14	20 / 100	1 ms	400 KiB

stove

#include <bits/stdc++.h>
#define int long long

using namespace std;

const int O = 2e5 + 5;
const int base = 500;
const int mod = 998244353;

int n, k, a[O];

main(){
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    cin >> n >> k;
    vector <int> b;

    for (int i = 1; i <= n; ++ i){
        cin >> a[i];
        if (i) b.push_back(a[i] - 1 - a[i - 1]);
    }

    sort(b.begin(), b.end());
    reverse(b.begin(), b.end());

    int res = 0;
    for (int i = 0; i < k - 1; ++ i){
        res += b[i];
    }

    cout << a[n] + 1 - a[1] - res;
}

Compilation message (stderr)

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

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