Submission #854340

#	Time	Username	Problem	Language	Result	Execution time	Memory
854340		dungz	Stove (JOI18_stove)	C++17	100 / 100	16 ms	2280 KiB

stove

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
#define fi first 
#define se second
#define endl '\n'
const ll MOD=1e9+7;
int a[100005];
vector<int> vec;
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		if(i>1) vec.push_back(a[i]-a[i-1]-1);
	}	
	ll tot=a[n]-a[1]+1;
	sort(vec.begin(),vec.end(),greater<int>());
	for(int i=0;i<min((int)(vec.size()),k-1);i++)
	{
		tot-=vec[i];
	}
	cout<<tot;
}

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