Submission #1049733

#	Time	Username	Problem	Language	Result	Execution time	Memory
1049733		vjudge1	Stove (JOI18_stove)	C++17	100 / 100	36 ms	1996 KiB

stove

#include <iostream>
#include <algorithm>
using namespace std;
long long ans ;
int main() 
{
  long long n,k;
  cin>>n>>k;

  ans = n;
  
  long long a[n];
  long long pr[n-1];
  long long pok = k;
  for(long long i = 0 ;i<n; i++){
    cin>>a[i];
  }
  for(long long i = 0; i<n-1; i++){
    pr[i]=a[i+1]-a[i]-1;
  }
  sort(pr,pr+n-1);
  
  for(int i = 0 ; i<n-1;i++){
    if(pr[i]==0){
      pok++;
      pr[i]=1000000001;
      if(pok>=n){
        break;
      }
    }
    else{
      break;
    }
  }
  if(pok<n){
    sort(pr,pr+n-1);
    for(int i = 0 ; i<n-pok; i++){
      ans+=pr[i];
    }
    cout<<ans;
  }
  else{
    cout<<ans;
  }
}

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