이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define FOR(i, a, b) for (int i=a; i<(b); i++)
#define F0R(i, a) for (int i=0; i<(a); i++)
#define FORd(i,a,b) for (int i = (b)-1; i >= a; i--)
#define F0Rd(i,a) for (int i = (a)-1; i >= 0; i--)
ll S[10005];
ll DP[10005][205];
const ll INF = 1e18;
void fill(int lev, int l, int r, int optl, int optr) {
if (l > r) return;
optr = min(optr,r);
int m = (l+r)/2;
DP[m][lev] = 1e18;
int bes = -1;
FOR(i,optl,min(optr+1,m)) {
if ((S[m] - S[i])*(S[m] - S[i])+DP[i][lev-1] < DP[m][lev]) {
DP[m][lev] = (S[m] - S[i])*(S[m] - S[i])+DP[i][lev-1];
bes = i;
}
}
fill(lev,l,m-1,optl,bes);
fill(lev,m+1,r,bes,optr);
}
int main(){
S[0] = 0;
int N, K; cin >> N >> K;
for (int i = 1; i <= N; i++){
ll x; cin >> x;
S[i] = S[i - 1] + x;
}
for (int i = 1; i <= N; i++){
DP[i][1] = S[i]*S[i];
}
for (int j = 2; j <= K + 1; j++){
fill(j, 1, N, 1, N);
}
int curpos = N;
vector<int> idx;
for (int j = K + 1; j >= 2; j--){
for (int i = curpos - 1; i >= 0; i--){
if (DP[curpos][j] == DP[i][j - 1] + (S[i] - S[curpos])*(S[i] - S[curpos])){
idx.push_back(i);
curpos = i;
break;
}
}
}
sort(idx.begin(), idx.end());
cout << (S[N] * S[N] - DP[N][K + 1])/2 << endl;
for (int j: idx){
cout << j << ' ';
}
cout << endl;
}
# | 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... |