Submission #1263641

#	Time	Username	Problem	Language	Result	Execution time	Memory
1263641		bachtd	Stove (JOI18_stove)	C++20	100 / 100	14 ms	1864 KiB

stove

/*#pragma GCC optimize(3)
#pragma GCC optimize("Ofast")
#pragma GCC optimize("inline")*/
#include <bits/stdc++.h>
#define int long long
#define Log2(x) 63 - __builtin_clzll(x)
#define pii pair<int, int>
#define se second
#define fi first
using namespace std;

const int N = 2e5 + 5;
const int inf = 1e18;
const int MOD = 1e9 + 7;
int n, m, LOG = 1;
int a[N], diff[N];

main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int k; cin >> n >> k;
    k--;
    for (int i=1; i<=n; ++i) cin >> a[i];
    sort(a+1, a+n+1);
    for (int i=1; i<n; ++i) {
        diff[i]=a[i+1]-a[i];
    }
    int sum=a[n]-a[1]+1;
    sort(diff+1, diff+n+1);
    reverse(diff+1, diff+n+1);
    for (int i=1; i<=k; ++i) {
        sum-=diff[i]-1;
    }
    cout << sum;
}

Compilation message (stderr)

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

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