Submission #1180643

#	Time	Username	Problem	Language	Result	Execution time	Memory
1180643		takoshanava	Stove (JOI18_stove)	C++20	100 / 100	30 ms	2632 KiB

stove

#include <bits/stdc++.h>
#define int long long
#define pb push_back
#define fs first
#define sc second
using namespace std;

const int N = 100000;
int a[N], b[N], diff[N];

signed main() {
	int n, k;
    cin >> n >> k;

    if(k >= n){
        cout << n << endl;
        return 0;
    }
    
    for(int i = 1; i <= n; i++){
        cin >> a[i];
        b[i] = a[i] + 1;
    }

    for(int i = 1; i < n; i++){
        diff[i] = a[i + 1] - b[i];
    }

    sort(diff + 1, diff + n, greater<int>());

    int cnt = 0, sum = 0;

    while(k--){
        sum += diff[cnt];
        cnt++;
    }

    cout << b[n] - a[1] - sum;
}

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