Submission #1254433

#	Time	Username	Problem	Language	Result	Execution time	Memory
1254433		hoa208	Stove (JOI18_stove)	C++20	100 / 100	12 ms	2244 KiB

stove

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i, a, b) for (int i = (a), _b = (b); i <= _b; i++)
#define FORD(i, b, a) for (int i = (b), _a = (a); i >= _a; i--)
#define pa pair<ll, ll>
#define fi first
#define se second
#define bit(mask, j) ((mask >> j) & 1)
#define t_test int t;cin >> t;while(t--)
const   ll mod = 1e9 + 7;
const   ll INF = 1e18;
inline  void adm(ll &x){if(x>=mod)x%=mod;else if(x<0)x+=mod;}
//--------------------------------------------------------------------

const ll N = 1e5 + 10;
ll n, k;
ll t[N];
void hbmt() {
    cin >> n >> k;
    FOR(i, 1, n) {
        cin >> t[i];
    }
    ll sum = t[n] - t[1] + 1;
    vector<ll> vt;
    FOR(i, 2, n) {
        ll x = t[i] - t[i - 1] - 1;
        vt.push_back(x);
    }
    sort(vt.begin(), vt.end(), greater<ll>());
    FOR(i, 0, min<ll>(vt.size(), k - 1) - 1) {
        sum -= vt[i];
    }
    cout << sum;
}

int main() {
    
    ios_base::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    
    if(fopen("hbmt.inp", "r")) {
        freopen("hbmt.inp", "r", stdin);
        freopen("hbmt.out", "w", stdout);
    }
    
    // t_test 
    hbmt();
    return 0;
}

Compilation message (stderr)

stove.cpp: In function 'int main()':
stove.cpp:43:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   43 |         freopen("hbmt.inp", "r", stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
stove.cpp:44:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   44 |         freopen("hbmt.out", "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

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