제출 #1049749

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1049749		vjudge1	Stove (JOI18_stove)	C++17	100 / 100	20 ms	3024 KiB

stove

#include <bits/stdc++.h>

#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,lzcnt,mmx,abm,avx,avx2,fma")
#pragma GCC optimize("Ofast,unroll-loops,no-stack-protector,fast-math,inline")

using namespace std;

#define speed ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0)
#define vc vector
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define ll long long
#define all(x) x.begin(),x.end()
#define sz(x) (int)x.size()
#define int ll

const int N = 1e5 + 5;
const int M = 1e5 + 5;
const int mod = 1e9 + 7;
const int block = 500;

int n, k, a[N];

void solve() {
	cin >> n >> k;
    ll ans = n;
    for ( int i = 1; i <= n; i++ ) {
        cin >> a[i];
    }
    vc <int> save, dist;
    sort(a+1,a+n+1);
    for ( int i = 2; i <= n; i++ ) {
        if ( a[i] - a[i-1] <= 1 ) continue;
        dist.pb(a[i]-a[i-1]-1);
        save.pb(a[i]);
    }
    save.pb(a[n]);
    if ( sz(save) <= k ) {
        cout << ans;
        return;
    }
    sort(all(dist));
    int k2 = sz(save) - k;
    for ( int i = 0; i < k2; i++ ) {
        ans += dist[i];
    }
    cout << ans;
}

signed main() {
	speed;
	int tt;
	//cin >> tt;
    tt = 1;
	while ( tt-- ) {
		solve();
		cout << '\n';
	}
}

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