제출 #1133443

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1133443		FZ_Laabidi	Stove (JOI18_stove)	C++20	100 / 100	30 ms	2240 KiB

stove

#include <bits/stdc++.h>//
#define int long long
using namespace std;

signed main() {

    int n, k; cin >> n >>k;
    vector <int> v(n), gaps;
    for(int i = 0; i < n; i++)cin >> v[i];
    sort(v.begin(), v.end());
    if (n==k)
        cout << k<< endl;
    else if (k==1)
        cout << v[n-1]-v[0]+1 << endl;
    else{
        int cur = 0;
        for (int i=1; i<n; i++) {
            if (v[i]-v[i-1]>1)
                gaps.push_back( v[i] - v[i-1]);
        }
        int l = gaps.size();
        sort(gaps.begin(), gaps.end());
        int tofil = l-k+1;
            if (l+1 > k)
            {

                for (int i=0; i<tofil; i++) {
                    cur+=gaps[i];
                    cur--;
                }
            }
            cout <<n+cur << endl;
        }
    }

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