Submission #384518

#	Time	Username	Problem	Language	Result	Execution time	Memory
384518		aryan12	Stove (JOI18_stove)	C++17	100 / 100	21 ms	2468 KiB

stove

#include <bits/stdc++.h>
using namespace std;
#define int long long

int32_t main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	int n, k;
	cin >> n >> k;
	vector<int> diff;
	int prev, cur;
	for(int i = 1; i <= n; i++) {
		cin >> cur;
		if(i != 1) {
			diff.push_back(cur - prev);
		}
		prev = cur;
	}
	sort(diff.begin(), diff.end());
	int ans = 0;
	for(int i = 0; i < diff.size() + 1 - k; i++) {
		ans += diff[i];
	}
	ans += k;
	cout << ans << "\n";
}

Compilation message (stderr)

stove.cpp: In function 'int32_t main()':
stove.cpp:21:19: warning: comparison of integer expressions of different signedness: 'long long int' and 'long long unsigned int' [-Wsign-compare]
   21 |  for(int i = 0; i < diff.size() + 1 - k; i++) {
      |                 ~~^~~~~~~~~~~~~~~~~~~~~

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