Submission #1138972

#	Time	Username	Problem	Language	Result	Execution time	Memory
1138972		AmirMakaM	Stove (JOI18_stove)	C++20	100 / 100	13 ms	1480 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define ull unsigned long long
#define ll long long
#define ld long double
#define pb push_back
#define f first
#define s second 
#define sz(x) (int)x.size()
#define all(x) x.begin(),x.end()
#define pii pair<int,int> 
#define pll pair<ll,ll>
#define pld pair<ld,ld>
#define pdd pair<double,double>
#define mp make_pair   
#define AmirMakaM ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0)
// solve it
const ull SEED = chrono::steady_clock::now().time_since_epoch().count();
mt19937_64 mrand(SEED);
ull rnd(ull x = ~(0ull)) {return mrand() % x;} 
const ll MOD = 998244353;
const ll INF = 1e16+20;
const int inf = 1e9 + 7;
const ll N = 2e5+5;
const ll M = 2e3+1;
const double pi = 2*acos(0.0);
const int dx[] = {1,-1,0,0}, dy[] = {0,0,1,-1};

void solve() {
    int n, k;
    cin >> n >> k;
    int t[n+1];
    for(int i=1; i<=n; i++) cin >> t[i];

    vector<int> q;
    for(int i=1; i<n; i++) {
        q.pb(t[i+1]-t[i]-1);
    }
    sort(all(q));
    reverse(all(q));

    int ans = t[n]+1-t[1];
    for(int i=1; i<=k-1; i++) ans -= q[i-1];
    cout << ans;
}

int main() {
    AmirMakaM;
    srand(SEED);
    //freopen("walk.in", "r", stdin);
    //freopen("walk.out", "w", stdout);
    int ttt = 1;
    //cin >> ttt;
    while(ttt--) {
        solve();
    }
    return 0;
}

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