Submission #1310658

#	Time	Username	Problem	Language	Result	Execution time	Memory
1310658		syanvu	Stove (JOI18_stove)	C++20	100 / 100	14 ms	2368 KiB

stove

// #pragma optimize ("g",on)
// #pragma GCC optimize ("inline")
// #pragma GCC optimize ("Ofast")
// #pragma GCC optimize ("unroll-loops")
// #pragma GCC optimize ("03")
#include <bits/stdc++.h>

#define pb push_back
#define SS ios_base::sync_with_stdio(0);cin.tie(nullptr);cout.tie(nullptr);
#define int long long
#define all(v) v.begin(),v.end()
using namespace std;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

const int N = 4e5 + 1, inf = 1e9 + 1, mod = 998244353;

void solve(){
    int n, k;
    cin >> n >> k;
    int t[n + 1];
    int ans = n;
    for(int i = 1; i <= n; i++){
        cin >> t[i];
    }
    vector<int> v;
    for(int i = 1; i < n; i++){
        v.push_back(t[i + 1] - t[i] - 1);
    }
    sort(all(v));
    k = n - k;
    for(int i = 0; i < k; i++){
        ans += v[i];
    }
    cout << ans;
}

signed main(){
    SS
    // freopen("trains.in", "r", stdin);
    // freopen("trains.out", "w", stdout);

    int t = 1;
    // cin >> t;
    while(t--){
        solve();
    }
}

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