//#pragma GCC optimize("O3")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,tune=native")
//#pragma GCC optimize ("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
const int mod = 1e9 + 7;
const int N = 1e5 + 5;
const int K = 205;
struct Line {
ll k, b;
int pos;
ll eval(ld x) {
return k * x + b;
}
};
inline ld isec(Line &p, Line &q) {
return ((ld)p.b - q.b) / (q.k - p.k);
}
struct CHT {
int s;
Line l[N];
int pt;
void clr() {
s = 0, pt = 0;
}
void add(Line a) {
while(s >= 2 && isec(l[s - 2], l[s - 1]) >= isec(l[s - 1], a)) s--, pt = min(pt, s - 1);
l[s++] = a;
}
pair<ll, int> get(ll x) {
while(pt != s - 1 && isec(l[pt], l[pt + 1]) <= x) pt++;
return { l[pt].eval(x), l[pt].pos };
}
};
int a[N];
ll pref[N];
ll dp[K][N];
int way[K][N];
CHT cht;
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int n, k;
cin >> n >> k;
for (int i = 1; i <= n; i++) {
cin >> a[i];
pref[i] = pref[i - 1] + a[i];
}
for (int i = 1; i <= k; i++) {
cht.clr();
for (int j = 1; j <= n; j++) {
if (j > i) {
auto p = cht.get(pref[j]);
dp[i][j] = p.first;
way[i][j] = p.second;
}
if (pref[j]) {
cht.add({ pref[j], dp[i - 1][j] - pref[j] * pref[j], j });
}
}
}
cout << dp[k][n] << '\n';
int kek = n;
vector<int> ans;
for (int i = k; i > 0; i--) {
ans.push_back(way[i][kek]);
kek = way[i][kek];
}
reverse(ans.begin(), ans.end());
for (int i : ans) {
cout << i << ' ';
}
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
2 ms |
376 KB |
contestant found the optimal answer: 108 == 108 |
2 |
Correct |
11 ms |
348 KB |
contestant found the optimal answer: 999 == 999 |
3 |
Incorrect |
2 ms |
376 KB |
Integer 0 violates the range [1, 1] |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
376 KB |
contestant didn't find the optimal answer: 252308 < 1093956 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
384 KB |
contestant didn't find the optimal answer: 484133 < 610590000 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
3 ms |
504 KB |
contestant didn't find the optimal answer: 395622 < 21503404 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
8 ms |
984 KB |
contestant didn't find the optimal answer: 1187850 < 1818678304 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
91 ms |
7416 KB |
contestant didn't find the optimal answer: 5054352 < 19795776960 |
2 |
Halted |
0 ms |
0 KB |
- |