#include <iostream>
#include <vector>
using namespace std;
using ll = long long;
int n, k;
vector <ll> a(1e5 + 1, 0);
vector <ll> p(1e5 + 1, 0);
vector <ll> dp(1e5 + 1, 0);
vector <ll> cnt(1e5 + 1, -1);
ll sum(int l, int r) {
return p[r] - p[l - 1];
}
pair <ll, ll> do_dp(ll lam) {
dp[0] = -lam;
int last = 0;
for (int i = 1; i <= n; i++) {
dp[i] = 1e18;
for (int j = last; j < i; j++) {
ll cur = dp[j] + lam + sum(j + 1, i) * (i - j);
if (cur > dp[i]) break;
dp[i] = cur;
cnt[i] = cnt[j] + 1;
last = j;
}
}
return {cnt[n], dp[n]};
}
int main() {
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> a[i];
p[i] = p[i - 1] + a[i];
}
ll l = 0, r = 1e18, m;
while (r - l > 1) {
m = (l + r) / 2;
(do_dp(m).first >= k ? l = m : r = m);
}
cout << do_dp(l).second - k * l;
}| Subtask | Sum | Test | Verdict | Time | Memory | ||
|---|---|---|---|---|---|---|---|
| subtask1 | 0/0 | ||||||
| 1 | Accepted | 4ms | 7900 KiB | ||||
| 2 | Wrong answer | 9ms | 8176 KiB | ||||
| subtask2 | 10/10 | ||||||
| 3 | Accepted | 4ms | 8472 KiB | ||||
| 4 | Accepted | 4ms | 8660 KiB | ||||
| 5 | Accepted | 4ms | 8872 KiB | ||||
| 6 | Accepted | 4ms | 8812 KiB | ||||
| 7 | Accepted | 4ms | 9068 KiB | ||||
| subtask3 | 0/10 | ||||||
| 8 | Accepted | 6ms | 9160 KiB | ||||
| 9 | Wrong answer | 4ms | 9116 KiB | ||||
| 10 | Wrong answer | 6ms | 9372 KiB | ||||
| 11 | Wrong answer | 6ms | 9692 KiB | ||||
| 12 | Wrong answer | 4ms | 9968 KiB | ||||
| subtask4 | 0/20 | ||||||
| 13 | Wrong answer | 8ms | 9720 KiB | ||||
| 14 | Wrong answer | 10ms | 9668 KiB | ||||
| 15 | Wrong answer | 10ms | 9668 KiB | ||||
| 16 | Wrong answer | 12ms | 9700 KiB | ||||
| 17 | Wrong answer | 12ms | 9724 KiB | ||||
| 18 | Wrong answer | 10ms | 9692 KiB | ||||
| subtask5 | 0/29 | ||||||
| 19 | Wrong answer | 57ms | 9948 KiB | ||||
| 20 | Wrong answer | 61ms | 9868 KiB | ||||
| 21 | Wrong answer | 68ms | 10000 KiB | ||||
| 22 | Wrong answer | 71ms | 10292 KiB | ||||
| 23 | Wrong answer | 72ms | 10236 KiB | ||||
| subtask6 | 0/31 | ||||||
| 24 | Accepted | 71ms | 10416 KiB | ||||
| 25 | Accepted | 75ms | 10168 KiB | ||||
| 26 | Wrong answer | 92ms | 10172 KiB | ||||
| 27 | Wrong answer | 98ms | 10136 KiB | ||||
| 28 | Wrong answer | 103ms | 10388 KiB | ||||
| 29 | Wrong answer | 108ms | 10656 KiB | ||||
| 30 | Wrong answer | 114ms | 10596 KiB | ||||
| 31 | Wrong answer | 125ms | 10632 KiB | ||||
| 32 | Wrong answer | 130ms | 10920 KiB | ||||
| 33 | Wrong answer | 131ms | 10852 KiB | ||||
| 34 | Wrong answer | 134ms | 10724 KiB | ||||
| 35 | Wrong answer | 141ms | 10728 KiB | ||||
| 36 | Wrong answer | 138ms | 10760 KiB | ||||
| 37 | Wrong answer | 136ms | 11020 KiB | ||||
| 38 | Wrong answer | 140ms | 11052 KiB | ||||
| 39 | Wrong answer | 133ms | 11088 KiB | ||||
| 40 | Wrong answer | 118ms | 10976 KiB | ||||