#include <bits/stdc++.h>
#pragma GCC optimize("O3,unroll-loops")
#pragma GCC target("avx2")
using namespace std;
int n,k;
int a[1000005];
int rmq[1000005][20];
int dp[2][1000005];
int curr = 1;
int get(int l,int r){
int v = __lg(r - l + 1);
return max(rmq[l][v],rmq[r - (1 << v) + 1][v]);
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin >> n >> k;
for(int i = 1;i <= n;i++){
cin >> a[i];
rmq[i][0] = a[i];
dp[0][i] = 0;
}
for(int k = 1;(1 << k) <= n;k++){
for(int i = 1;i + (1 << k) - 1 <= n;i++){
rmq[i][k] = max(rmq[i][k - 1],rmq[i + (1 << (k - 1))][k - 1]);
}
}
for(int i = 1;i <= n;i++){
dp[0][i] = 0;
}
for(int j = 1;j <= k;j++){
//cout << j << " : \n";
for(int i = j;i <= n;i++){
dp[curr][i] = dp[1 - curr][i - 1] + a[i];
for(int ii = i - 2;ii >= j - 1;ii--){
dp[curr][i] = min(dp[curr][i],dp[1 - curr][ii] + get(ii + 1,i));
}
}
curr = (curr + 1) % 2;
/*for(int i = 1;i <= n;i++) cout << dp[1 - curr][i] << " ";
cout << "\n";
for(int i = 1;i <= n;i++) cout << dp[curr][i] << " ";
cout << "\n";*/
}
cout << dp[1 - curr][n];
return 0;
}
# | 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... |