Submission #1007345

#	Time	Username	Problem	Language	Result	Execution time	Memory
1007345		em4ma2	Stove (JOI18_stove)	C++14	100 / 100	13 ms	2392 KiB

stove

#include <bits/stdc++.h>
 
using namespace std;
 
#define applejuice ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define mod 1000000007
#define endl '\n'
#define mxsz 100005
#define ll long long
 
signed main(){
    
    applejuice;
    
    int n,k;
    cin>>n>>k;
    int a[n];
    for (int i=0;i<n;i++)cin>>a[i];
    
    int diff[n];
    diff[0]=1;
    
    for (int i=1;i<n;i++){
        diff[i]=a[i]-a[i-1];
    }
    
    sort(diff,diff+n);
    reverse(diff,diff+n);
    
    k--;
    
    for (int i=0;i<n;i++){
        if (k==0)break;
        
        if (diff[i]>1){
            diff[i]=1;
            k--;
        }
    }
    
    int sum=0;
    
    for (int i=0;i<n;i++){
        sum+=diff[i];
    }
    
    cout<<sum;
    
    return 0;
}

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