Submission #474706

#	Time	Username	Problem	Language	Result	Execution time	Memory
474706		irmuun	Stove (JOI18_stove)	C++17	100 / 100	66 ms	3052 KiB

stove

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
ll p[100001],a,b,ans,c,i;
vector<ll>v;
int main(){
	cin>>a>>b;
	for(i=1;i<=a;i++){
		cin>>p[i];
		if(i>1){
			v.pb(p[i]-p[i-1]-1);
		}
	}
	ans=p[a]-p[1]+1;
	if(a<=b){
		cout<<a<<endl;
		v.clear();
	}
	else{
		b--;
		sort(v.begin(),v.end());
		c=a-2;
		for(i=1;i<=b;i++){
			ans-=v[c];
			c--;
		}
		cout<<ans<<endl;
	}
}

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