Submission #161514

#	Time	Username	Problem	Language	Result	Execution time	Memory
161514		kostia244	Stove (JOI18_stove)	C++14	100 / 100	37 ms	3316 KiB

stove

#pragma comment(linker, "/stack:200000000")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#pragma GCC optimize("unroll-loops")
#include<bits/stdc++.h>
#include<bits/extc++.h>
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define pb push_back
using namespace std;
//using namespace __gnu_pbds;
using ll = long long;
using vi = vector<ll>;
using vvi = vector<vector<ll>>;
const ll mod = 1e9 + 7;
//using oset = tree<pair<int, int>, null_type, less<pair<int, int>>, rb_tree_tag, tree_order_statistics_node_update>;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	ll n, k;
	vi a;
	cin >> n >> k;
	a.resize(n);
	for(auto &i : a) cin >> i;
	ll ans = a.back()-a[0]+1;
	vi seg;
	for(int i = 1; i < n; i++)
		if(a[i]-a[i-1]-1>0)
			seg.pb(a[i]-a[i-1]-1);
	sort(rall(seg));
	for(int i = 0; i < min((ll)seg.size(), k-1); i++)
		ans -= seg[i];
	cout << ans;
}

Compilation message (stderr)

stove.cpp:1:0: warning: ignoring #pragma comment  [-Wunknown-pragmas]
 #pragma comment(linker, "/stack:200000000")

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