# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
488139 | ntabc05101 | 수열 (APIO14_sequence) | C++14 | 26 ms | 11928 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include<bits/stdc++.h>
using namespace std;
#define taskname ""
int main() {
if (fopen(taskname".inp", "r")) {
freopen(taskname".inp", "r", stdin);
freopen(taskname".out", "w", stdout);
}
else {
if (fopen(taskname".in", "r")) {
freopen(taskname".in", "r", stdin);
freopen(taskname".out", "w", stdout);
}
}
cin.tie(0)->sync_with_stdio(0);
int n, K; cin >> n >> K;
long long a[n + 1];
a[0] = 0;
for (int i = 1; i <= n; i++) {
cin >> a[i]; a[i] += a[i - 1];
}
vector<long long> dp[n + 1];
function<double(int, int, int)> slope = [&](int k, int i, int j) {
return (double)(dp[i][k] - a[i] * a[i] - dp[j][k] + a[j] * a[j]) / (a[j] - a[i]);
};
K++;
int prf[n + 1][K + 1];
long long res = 0;
for (int i = 0; i <= n; i++) {
dp[i].assign(K + 1, 0);
}
deque<int> deq[K + 1];
for (int i = 0; i <= K; i++) {
deq[i].clear();
}
deq[0].push_back(0);
for (int i = 1; i <= n; i++) {
for (int k = 1; k <= K; k++) {
while (deq[k - 1].size() > 1 && slope(k - 1, deq[k - 1][0], deq[k - 1][1]) <= a[i]) {
deq[k - 1].pop_front();
}
if (deq[k - 1].empty()) {
continue;
}
int j = deq[k - 1].front();
//cout << i << " " << j << " " << k << " " << "\n";
dp[i][k] = dp[j][k - 1] + (a[i] - a[j]) * a[j];
prf[i][k] = j;
}
for (int k = 1; k <= K; k++) {
while (deq[k].size() > 1 && slope(k, deq[k][deq[k].size() - 2], deq[k].back()) >= slope(k, deq[k].back(), i)) {
deq[k].pop_back();
}
deq[k].push_back(i);
}
}
cout << "\n";
res = max(res, dp[n][K]);
cout << res << "\n";
while (K > 1) {
cout << prf[n][K] << " ";
n = prf[n][K--];
}
cout << "\n";
return 0;
}
/*
7 3
4 1 3 4 0 2 3
*/
컴파일 시 표준 에러 (stderr) 메시지
# | 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... |