제출 #388377

#제출 시각아이디문제언어결과실행 시간메모리
388377warner1129Stove (JOI18_stove)C++17
100 / 100
26 ms988 KiB
#include <bits/stdc++.h>

using namespace std;

const int maxn = 1e5 + 5;

int arr[maxn];
int n, m;

pair<int, int> DP(int p) {
	static int dp[maxn] = {};
	static int cnt[maxn] = {};
	int pre = 0;
	for (int i = 1; i <= n; i++) {
		dp[i] = dp[pre] + (arr[i] - arr[pre+1] + 1) + p;
		cnt[i] = cnt[pre] + 1;
		if (dp[i] - arr[i+1] < dp[pre] - arr[pre+1]) pre = i;
	}
	return {dp[n] - cnt[n] * p, cnt[n]};
}

void solve() {
	cin >> n >> m;
	for (int i = 1; i <= n; i++) cin >> arr[i];
	m = min(m, n);
	int l = 0, r = arr[n];
	while (l <= r) {
		int v, c;
		int mid = (l+r)>>1;
		tie(v, c) = DP(mid);
		if (c <= m) r = mid-1;
		else l = mid+1;
	}
	long long lv, lc, rv, rc;
	tie(lv, lc) = DP(l);
	tie(rv, rc) = DP(r);
	if (lc == rc) cout << lv << '\n';
	else cout << lv + (rv - lv) * (m - lc) / (rc - lc) << '\n';	
}

void solve2() {
	int n, m;
	priority_queue<int> pq;
	cin >> n >> m;
	int last, ans;
	cin >> last;
	ans = -last + 1;
	for (int i = 1; i < n; i++) {
		int x;
		cin >> x;
		pq.emplace(x - last - 1);
		last = x;
	}
	ans += last;
	int cnt = 1;
	m = min(m, n);
	while (cnt + 1 <= m) {
		ans -= pq.top();
		pq.pop();
		cnt++;
	}
	cout << ans << '\n';
}

signed main() {
	ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
	solve2();
	return 0;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...