Submission #1114741

#	Time	Username	Problem	Language	Result	Execution time	Memory
1114741		Vidicy	Stove (JOI18_stove)	C++17	100 / 100	18 ms	2896 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fr first
#define sec second

signed main() {
	ios::sync_with_stdio(NULL);
	cin.tie(NULL);
	cout.tie(NULL);
	
	int n,k;
	cin >> n >> k;
	int cnt = n-(n-k);
	vector<int> a(n);
	for(int i = 0; i < n; i ++)cin >> a[i];
	if(n == k) cout << k;
	else{	
		vector<int> b(n-1);
		for(int i = 0; i < n-1; i++){
			b[i] = abs(a[i] - a[i+1]);
		}
		
		sort(b.begin(),b.end());
		for(int i = 0; i < n-k; i++){
			cnt = cnt + b[i];
		}
		cout << cnt << endl;
	}
}

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