This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define all(a) a.begin(),a.end()
#define mp make_pair
#define fi first
#define se second
#define sz(a) (int)a.size()
const int maxn = (int)1e4+10;
const int INF = (int)2e18;
int n, K, a[maxn], s[maxn];
int dp[maxn][201], p[maxn][201];
vector<int> v;
void recur(int i, int k){ if(k) v.pb(i), recur(p[i][k],k-1); }
double slope(int i, int j, int k){
if(s[j]==s[i]) return INF;
return (double)((dp[j][k]-s[n]*s[j])-(dp[i][k]-s[n]*s[i]))/(s[j]-s[i]);
}
int32_t main()
{
ios_base::sync_with_stdio(false); cin.tie(0);
cin >> n >> K;
for(int i = 0; i < n; i++) cin >> a[i], s[i+1]=s[i]+a[i];
for(int k = 1; k <= K+1; k++){
deque<int> DQ; DQ.clear(); DQ.push_back(0);
for(int i = 1; i <= n; i++){
while(sz(DQ)>1 and slope(DQ[0],DQ[1],k-1)>= -s[i]) DQ.pop_front();
int j = DQ[0]; dp[i][k]=dp[j][k-1]+(s[i]-s[j])*(s[n]-s[i]), p[i][k]=j;
while(sz(DQ)>1 and slope(DQ[sz(DQ)-2], DQ.back(),k-1) <= slope(DQ.back(),i,k-1)) DQ.pop_back();
DQ.push_back(i);
}
}
cout << dp[n][K+1] << "\n"; recur(n,K+1);
sort(all(v)); v.pop_back(); for(auto u : v) cout << u << " ";
}
# | 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... |