이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
const int mxN = 1e5+2, mxK = 202;
long long pref[mxN], dp[mxK][mxN], to[mxK][mxN], c, t, h[mxN];
long double cord(int x, int y) {
return (long double)(dp[c][y] - pref[y]*pref[y] + pref[x]*pref[x] - dp[c][x])/(pref[x] - pref[y]); // deljenje 0-om?
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
int n, k;
cin >> n >> k;
vector<int> z;
for (int i = 1; i <= n; ++i) {
int x;
cin >> x;
z.push_back(i);
if (!x) {
h[t]++;
z.pop_back();
z.push_back(i-1);
continue;
}
z.pop_back();
++t;
h[t] = i;
pref[t] = pref[t-1] + x;
}
n = t;
if (n == 1) {
cout << 0 << '\n';
for (int i = 1; i <= k; ++i) {
cout << i << ' ';
}
return 0;
}
for (int i = 1; i <= k; ++i) {
deque<int> dq = {0};
for (int j = 1; j <= n; ++j) {
while (dq.size() > 1 && cord(dq[0], dq[1]) <= pref[j]) dq.pop_front();
dp[i][j] = pref[j]*pref[dq[0]] + dp[i-1][dq[0]] - pref[dq[0]]*pref[dq[0]];
to[i][j] = dq[0];
while (dq.size() > 1 && cord(dq[dq.size()-1], dq[dq.size()-2]) >= cord(dq[dq.size()-2], j)) dq.pop_back();
dq.push_back(j);
}
++c;
}
cout << dp[k][n] << '\n';
vector<int> v = {n};
int cnt = 0;
for (int i = k; i >= 1; --i) {
if (!to[i][v.back()]) {
cnt = i;
break;
}
v.push_back(to[i][v.back()]);
}
reverse(v.begin(), v.end()); v.pop_back();
for (auto i : v) {
cout << h[i] << ' ';
}
v.clear();
while (v.size() < cnt) {
v.push_back(z.back()); z.pop_back();
}
for (auto i : v) cout << i << ' ';
}
컴파일 시 표준 에러 (stderr) 메시지
sequence.cpp: In function 'int main()':
sequence.cpp:66:21: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
66 | while (v.size() < cnt) {
| ~~~~~~~~~^~~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |