제출 #337107

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
337107		beepbeepsheep	Stove (JOI18_stove)	C++17	100 / 100	21 ms	1028 KiB

stove

#include <bits/stdc++.h>

using namespace std;
#define ll long long

const ll bignum=1e17;
const int inf= 2e9;
const int mod= 1e9+7;

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    ll n,k,prev,curr;
    cin>>n>>k;
    vector<int> dis;
    cin>>prev;
    if (n==1){
        cout<<1;
        return 0;
    }
    ll ans=-prev;
    for (int i=1;i<n;i++){
        cin>>curr;
        dis.push_back(curr-prev-1);
        prev=curr;
    }
    ans+=curr+1;
    sort(dis.begin(),dis.end());
    for (int i=0;i<k-1;i++){
        ans-=dis[n-i-2];
    }
    cout<<ans;
    return 0;
}

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