제출 #1269164

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1269164		picrad	Stove (JOI18_stove)	C++20	100 / 100	14 ms	1992 KiB

stove

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
using namespace std;
typedef long long ll;
typedef double dbl;
typedef pair<ll,ll> pii;

const int maxn = 1e5+5;
ll N,K,T[maxn],ans;

int main(){
    cin.tie(0);
    ios_base::sync_with_stdio(0);
    cin >> N >> K;
    vector<ll> pq;
    for(int i =1; i <= N; i++){
        cin >> T[i];
        if(i > 1)pq.pb(T[i]-T[i-1]-1);
    }
    ans = T[N] - T[1] + 1;
    //cout << ans << '\n';
    sort(pq.begin(),pq.end());
    K--;
    while(K-- && !pq.empty()){
        ll cur = pq.back();
        pq.pop_back();
        ans -= cur;
    }
    cout << ans << endl;
}

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