Submission #792451

#	Time	Username	Problem	Language	Result	Execution time	Memory
792451		vjudge1	Stove (JOI18_stove)	C++17	100 / 100	17 ms	1620 KiB

stove

#ifdef Home
#define _GLIBCXX_DEBUG
#endif // Home

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;

main() {
#ifdef Home
    freopen("input.txt", "r", stdin);
    freopen("output.txt", "w", stdout);
#endif // Home
    ios_base::sync_with_stdio(0);
    cin.tie(0);

    int n, k, pr, cr, ans;
    cin >> n >> k >> pr;
    if(n == 1) {
        cout << 1;
        return 0;
    }
    vector < int > a(n);
    ans = pr;
    for(int i = 1; i < n; ++ i) {
        cin >> cr;
        a[i] = cr - pr;
        pr = cr;
    }
    pr = ans;
    ans = cr - ans + 1;
    sort(a.rbegin(), a.rend());
    for(int i = 0; i < k - 1; ++ i) {
        ans -= a[i] - 1;
    }
    cout << ans;
}

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: 30 / 30
• Subtask 3: 50 / 50