Submission #1150897

#	Time	Username	Problem	Language	Result	Execution time	Memory
1150897		danglayloi1	Stove (JOI18_stove)	C++20	100 / 100	21 ms	1480 KiB

stove

#include <bits/stdc++.h>
#define ii pair<int, int>
#define fi first
#define se second
#define inf 0x3f3f3f3f3f3f3f3f
using namespace std;
using ll = long long;
const ll mod=1e9+7;
const int nx=1e5+5;
int n, k, a[nx];
vector<int> len;
ll ans;
int main()
{
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin>>n>>k;
    for(int i = 1; i <= n; i++)
        cin>>a[i];
    if(n==1) return cout<<k, 0;
    sort(a+1, a+n+1);
    ans=a[n]-a[1]+1;
    for(int i = 1; i < n; i++)
        len.emplace_back(a[i+1]-a[i]-1);
    sort(len.begin(), len.end(), greater<int>());
    for(int i = 0; i < k-1; i++)
        ans-=len[i];
    cout<<ans;
}

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