Submission #1229051

#	Time	Username	Problem	Language	Result	Execution time	Memory
1229051		tutientd	Stove (JOI18_stove)	C++20	0 / 100	0 ms	324 KiB

stove

#include<bits/stdc++.h>
#define int long long
#define ff(i,a,b) for(int i=a;i<=b;i++)
#define fb(i,a,b) for(int i=a;i>=b;i--)
#define all(v) v.begin(),v.end()
#define pushb push_back
#define DEBUG cout<<"nO ERROR";return;
#define re return
using namespace std;
void faster(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
}
const int N=1e5+5;const  int INF=1e18;
 int n,k;
int32_t main() {
    faster();
    cin >> n >> k;
    vector<int> t(n);
    for (int i = 0; i < n; ++i) cin >> t[i];
    vector<int> gaps;
    ff(i,0,n-2){
        int gap = t[i + 1] - t[i] - 1;
        if(gap > 0) gaps.push_back(gap);
    }
    int ans=n;
    sort(gaps.begin(), gaps.end());
    for (int i = 0; i < n - k && i < (int)gaps.size(); ++i) {
        ans += gaps[i];
    }

    cout <<ans;
}

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