제출 #1290442

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1290442		mhaer	Stove (JOI18_stove)	C++20	100 / 100	16 ms	2372 KiB

stove

#include <bits/stdc++.h>
#define int long long
#define pii pair<int, int>
#define vc vector
#define Fin(a) for (auto& i : a)
#define fih ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define mp(a, b) make_pair(a, b)
#define pb(a) push_back(a)

//#pragma GCC optimize("unroll-loops")
#pragma GCC optimize("Ofast") 
//#pragma GCC target("avx,avx2,fma")
using namespace std;

signed main() {
    fih
    int n, k;
    cin >> n >> k;
    vector<int> input(n);
    for (int i = 0;i < n;i++) cin >> input[i];

    int mm = input[n-1] - input[0] + 1;

    vc<int> bos;
    for (int i = 0;i < n-1;i++) {
        bos.pb(input[i+1] - input[i] - 1);
    }

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

    while (--k && bos.size() > 0) {
        mm -= bos.back();
        bos.pop_back();
    }

    cout << mm << "\n";
}
 

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