#include "bits/stdc++.h"
using namespace std;
#define ll long long
#define forn(i,n) for(int i=0;i<n;i++)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(),v.rend()
#define pb push_back
#define sz(a) (int)a.size()
const int N = 1e3 + 10, K = 205;
ll dp[N][K], par[N][K];
ll p[N];
ll calc(int l, int r){
ll ret = p[r];
if(l)ret -= p[l - 1];
return ret;
}
void solve()
{
forn(i, N)forn(j, K)dp[i][j] = -1e7, par[i][j] = -1;
int n, k;
cin >> n >> k;
vector<int> a(n);
for(int i = 0;i < n; ++i){
cin >> a[i];
p[i] = a[i];
if(i)p[i] += p[i - 1];
}
forn(i, n)dp[i][0] = dp[0][i] = 0;
for(int i = 1; i < n; ++i){
for(int j = 0; j <= k; ++j){
dp[i][j] = 0;
if(j > 0){
for(int l = 0;l < i; ++l){
if(i + 1 < n){
if(dp[i][j] < dp[l][j - 1] + calc(l, i - 1) * calc(i, n - 1)){
par[i][j] = l;
}
dp[i][j] = max(dp[i][j], dp[l][j - 1] + calc(l, i - 1) * calc(i, n - 1));
}
}
}
}
}
ll mx = 0, idx = 0;
for(int i = 0;i < n; ++i){
if(dp[i][k] > mx){
mx = dp[i][k];
idx = i;
}
}
cout << mx << "\n";
int x = idx;
cout << x + 1 << " ";
x = par[idx][k];
for(int i = 1;i < k; ++i){
cout << x << " ";
x = par[x][k - i];
}
}
int32_t main()
{
ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
int t = 1;
//cin >> t;
while(t--)
{
solve();
}
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
2 ms |
3532 KB |
contestant found the optimal answer: 108 == 108 |
2 |
Incorrect |
2 ms |
3560 KB |
declared answer doesn't correspond to the split scheme: declared = 999, real = 900 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
3556 KB |
declared answer doesn't correspond to the split scheme: declared = 1064061, real = 1093956 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
3532 KB |
contestant didn't find the optimal answer: 514120000 < 610590000 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
7 ms |
3568 KB |
contestant found the optimal answer: 21503404 == 21503404 |
2 |
Incorrect |
5 ms |
3568 KB |
declared answer doesn't correspond to the split scheme: declared = 140412195, real = 140412192 |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
7 ms |
7128 KB |
Execution killed with signal 11 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
12 ms |
7808 KB |
Execution killed with signal 11 |
2 |
Halted |
0 ms |
0 KB |
- |