Submission #1290445

#	Time	Username	Problem	Language	Result	Execution time	Memory
1290445		dodopasa	Stove (JOI18_stove)	C++20	0 / 100	1 ms	568 KiB

stove

#include <bits/stdc++.h>
#define int long long
//#define cout cout << "Test: "
int sum=0;
using namespace std;
void solve(){
    int n,k;cin>>n>>k;
    vector<int> dp(n);
    vector<int> v(n);
    vector<int> ara(n-1);
    
    for(int i=0;i<n;i++){
        cin>>v[i];
        if(i==0){
            dp[0]=v[0];
            continue;
        }
        //cout<<"bbbbbbb";
        dp[i]=dp[i-1]+v[i];
        ara[i-1]=v[i]-v[i-1]-1;
    }
    if(n==k){
        cout<<n;
        return;
    }
    if(k==1){
        cout<<v[n-1];
        return;
    }
    sort(ara.begin(),ara.end());
    for(int i=n-2;i>k-2;i--){
        //cout<<"aaaaaa";
        sum+=ara[i];
    }
    cout<<v[n-1]-sum;
}

signed main(){
    //ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
    
    int t=1;
    //cin>>t;
    while(t--){
        solve();
        cout<<endl;
    }
    return 0;
}

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