# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
939439 | biximo | 수열 (APIO14_sequence) | Java | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define N 100005
using namespace std;
typedef long long ll;
struct Line {
ll m,b;
ll get(ll x) {
return m*x+b;
}
long double cross(Line v) {
return 1LL*(b-v.b)/(v.m-m);
}
};
struct CHT {
Line vt[N];
int ind[N];
int ft=0,bk=0;
void adds(Line v, int i) {
while(bk-ft) {
if(vt[bk-1].m == v.m) {
if(v.b < vt[bk-1].b) {
i = ind[bk-1];
v.b = vt[bk-1].b;
}
--bk;
} else {
break;
}
}
while(bk-ft >= 2) {
if(vt[bk-2].cross(v) < vt[bk-2].cross(vt[bk-1])) {
--bk;
} else {
break;
}
}
ind[bk] = i;
vt[bk++] = v;
}
ll query(int x) {
if(bk==ft) return 0;
while(bk-ft>=2) {
if(vt[ft].cross(vt[ft+1]) < x) {
++ft;
} else {
break;
}
}
return vt[ft].get(x);
}
void inits() {
ft = 0, bk = 0;
}
CHT() {}
};
ll n, k, pref[N], dp[N][2];
int bt[N][205];
CHT cht[2];
int main() {
cin.tie(0)->sync_with_stdio(0);
cin >> n >> k;
for(int i = 1; i <= n; i ++) {
cin >> pref[i];
pref[i] += pref[i-1];
}
cht[1].adds({0,0},0);
ll ans = -1, ind;
for(int j = 1; j <= k; j ++) {
swap(cht[0],cht[1]);
cht[1].inits();
for(int i = 1; i < j; i ++) {
swap(dp[i][0],dp[i][1]);
cht[0].adds({pref[i],-pref[n]*pref[i]+dp[i][0]},i);
}
for(int i = j; i <= n; i ++) {
swap(dp[i][0],dp[i][1]);
dp[i][1] = cht[0].query(pref[i])+pref[n]*pref[i]-pref[i]*pref[i];
bt[i][j] = cht[0].ind[cht[0].ft];
if(j == k && ans < dp[i][1]) {
ind = i;
ans = dp[i][1];
}
if(j > 1) {
cht[0].adds({pref[i],-pref[n]*pref[i]+dp[i][0]},i);
}
}
}
vector<int> pt;
int K = k;
while(ind) {
pt.push_back(ind);
ind = bt[ind][K--];
}
cout << ans << "\n";
sort(pt.begin(),pt.end());
for(int i: pt) cout << i << " ";
}