#include <cstdio>
#include <climits>
#include <functional>
#include <numeric>
#include <cstring>
#include <cassert>
#include <string>
#include <deque>
#include <vector>
#include <map>
#include <queue>
#include <algorithm>
#include <iostream>
#include <utility>
using namespace std;
using ll=long long;
#define N 300005
#define ALL(x) x.begin(), x.end()
int n, k, a[N];
long long p[N];
int main()
{
cin.tie(0)->sync_with_stdio(0);
cin >> n >> k;
for (int i = 1; i <= n; ++i) cin >> a[i], p[i] = p[i-1] + a[i];
if (count_if(a+1, a+1+n,[](int x){ return x<0; }) == 0) return cout << accumulate(a+1,a+1+n, 0ll), 0;
if (count_if(a+1, a+1+n,[](int x){ return x<0; }) == 1)
{
int x=0;
for (int i = 1; i <= n; ++i) if (a[i] < 0) x = i;
if (k == 1) return cout << max({p[x-1], p[n] - p[x], p[n]}), 0;
else return cout << p[n] - a[x], 0;
}
if (k == 1)
{
long long ans = 0, mn = 0;
for (int i = 1; i <= n; ++i) ans = max(ans, p[i] - mn), mn = min(mn, p[i]);
return cout << ans, 0;
}
if (n <= 300)
{
long long dp[305][305]{0};
for (int i = 1; i <= n; ++i)
for (int j = 1; j <= k; ++j)
{
dp[i][j] = max({dp[i-1][j], dp[i][j], dp[i][j-1]});
for (int l = 0; l < i; ++l) dp[i][j] = max(dp[i][j], dp[l][j-1] + p[i] - p[l]);
}
return cout << dp[n][k], 0;
}
if (n <= 3000)
{
vector<vector<vector<long long>>> dp(3005, vector<vector<long long>>(3005, vector<long long>(2)));
long long ans = 0;
for (int i = 1; i <= n; ++i)
{
for (int j = 1; j <= k; ++i)
{
dp[i][j][0] = max(dp[i-1][j][0], dp[i-1][j][1]);
dp[i][j][1] = max({a[i] + dp[i-1][j-1][0], a[i] + dp[i-1][j][1]});
ans = max({ans, dp[i][j][0], dp[i][j][1]});
}
}
return cout << ans, 0;
}
//assert(0);
#if 0
{
function<void(int, int, int, int, int)> compute = [&](int i, int l, int r, int optl, int optr)
{
if (l > r) return;
int m = (l+r)/2;
pair<long long, int> best = {LLONG_MIN, -1};
for (int k = optl; k <= min(m, optr); ++k)
best = max(best, {(dp[!(i&1)][k]) + p[m] - p[k], k});
dp[i&1][m] = best.first;
compute(i, l, m-1, optl, best.second), compute(i, m+1, r, best.second, optr);
};
long long ans = 0;
for (int i = 1; i <= k; ++i)
{
compute(i, 1, n, 0, n-1);
for (int i = 1; i <= n; ++i) dp[i&1][i] = max(dp[i&1][i], dp[i&1][i-1]);
ans = max(ans, *max_element(dp[i&1], dp[i&1]+n+1));
}
return cout << ans,0;
}
#endif
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
23 ms |
6096 KB |
Output is correct |
2 |
Correct |
24 ms |
6304 KB |
Output is correct |
3 |
Correct |
25 ms |
6252 KB |
Output is correct |
4 |
Correct |
24 ms |
6480 KB |
Output is correct |
5 |
Correct |
24 ms |
6228 KB |
Output is correct |
6 |
Correct |
24 ms |
5972 KB |
Output is correct |
7 |
Correct |
23 ms |
5972 KB |
Output is correct |
8 |
Correct |
23 ms |
6440 KB |
Output is correct |
9 |
Correct |
26 ms |
5968 KB |
Output is correct |
10 |
Correct |
25 ms |
6276 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
15 ms |
5224 KB |
Output is correct |
2 |
Correct |
15 ms |
5468 KB |
Output is correct |
3 |
Correct |
15 ms |
5192 KB |
Output is correct |
4 |
Correct |
15 ms |
5212 KB |
Output is correct |
5 |
Correct |
24 ms |
6080 KB |
Output is correct |
6 |
Correct |
16 ms |
5200 KB |
Output is correct |
7 |
Correct |
16 ms |
5468 KB |
Output is correct |
8 |
Correct |
24 ms |
6236 KB |
Output is correct |
9 |
Correct |
23 ms |
6096 KB |
Output is correct |
10 |
Correct |
16 ms |
5236 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
26 ms |
6340 KB |
Output is correct |
2 |
Correct |
26 ms |
6232 KB |
Output is correct |
3 |
Correct |
26 ms |
6228 KB |
Output is correct |
4 |
Correct |
26 ms |
6268 KB |
Output is correct |
5 |
Correct |
26 ms |
6436 KB |
Output is correct |
6 |
Correct |
26 ms |
6228 KB |
Output is correct |
7 |
Correct |
35 ms |
6404 KB |
Output is correct |
8 |
Correct |
26 ms |
6232 KB |
Output is correct |
9 |
Correct |
27 ms |
6228 KB |
Output is correct |
10 |
Correct |
26 ms |
6276 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1112 KB |
Output is correct |
2 |
Correct |
1 ms |
1116 KB |
Output is correct |
3 |
Correct |
1 ms |
1116 KB |
Output is correct |
4 |
Correct |
1 ms |
1116 KB |
Output is correct |
5 |
Correct |
1 ms |
1116 KB |
Output is correct |
6 |
Correct |
1 ms |
1128 KB |
Output is correct |
7 |
Correct |
1 ms |
1116 KB |
Output is correct |
8 |
Correct |
1 ms |
1112 KB |
Output is correct |
9 |
Correct |
1 ms |
1116 KB |
Output is correct |
10 |
Correct |
1 ms |
1116 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1112 KB |
Output is correct |
2 |
Correct |
1 ms |
1116 KB |
Output is correct |
3 |
Correct |
1 ms |
1116 KB |
Output is correct |
4 |
Correct |
1 ms |
1116 KB |
Output is correct |
5 |
Correct |
1 ms |
1116 KB |
Output is correct |
6 |
Correct |
1 ms |
1128 KB |
Output is correct |
7 |
Correct |
1 ms |
1116 KB |
Output is correct |
8 |
Correct |
1 ms |
1112 KB |
Output is correct |
9 |
Correct |
1 ms |
1116 KB |
Output is correct |
10 |
Correct |
1 ms |
1116 KB |
Output is correct |
11 |
Correct |
3 ms |
1116 KB |
Output is correct |
12 |
Correct |
3 ms |
1116 KB |
Output is correct |
13 |
Correct |
2 ms |
1116 KB |
Output is correct |
14 |
Correct |
3 ms |
1116 KB |
Output is correct |
15 |
Correct |
2 ms |
1116 KB |
Output is correct |
16 |
Correct |
2 ms |
1116 KB |
Output is correct |
17 |
Correct |
3 ms |
1116 KB |
Output is correct |
18 |
Correct |
1 ms |
1116 KB |
Output is correct |
19 |
Correct |
1 ms |
1116 KB |
Output is correct |
20 |
Correct |
1 ms |
1116 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
1112 KB |
Output is correct |
2 |
Correct |
1 ms |
1116 KB |
Output is correct |
3 |
Correct |
1 ms |
1116 KB |
Output is correct |
4 |
Correct |
1 ms |
1116 KB |
Output is correct |
5 |
Correct |
1 ms |
1116 KB |
Output is correct |
6 |
Correct |
1 ms |
1128 KB |
Output is correct |
7 |
Correct |
1 ms |
1116 KB |
Output is correct |
8 |
Correct |
1 ms |
1112 KB |
Output is correct |
9 |
Correct |
1 ms |
1116 KB |
Output is correct |
10 |
Correct |
1 ms |
1116 KB |
Output is correct |
11 |
Correct |
3 ms |
1116 KB |
Output is correct |
12 |
Correct |
3 ms |
1116 KB |
Output is correct |
13 |
Correct |
2 ms |
1116 KB |
Output is correct |
14 |
Correct |
3 ms |
1116 KB |
Output is correct |
15 |
Correct |
2 ms |
1116 KB |
Output is correct |
16 |
Correct |
2 ms |
1116 KB |
Output is correct |
17 |
Correct |
3 ms |
1116 KB |
Output is correct |
18 |
Correct |
1 ms |
1116 KB |
Output is correct |
19 |
Correct |
1 ms |
1116 KB |
Output is correct |
20 |
Correct |
1 ms |
1116 KB |
Output is correct |
21 |
Runtime error |
182 ms |
262144 KB |
Execution killed with signal 9 |
22 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
23 ms |
6096 KB |
Output is correct |
2 |
Correct |
24 ms |
6304 KB |
Output is correct |
3 |
Correct |
25 ms |
6252 KB |
Output is correct |
4 |
Correct |
24 ms |
6480 KB |
Output is correct |
5 |
Correct |
24 ms |
6228 KB |
Output is correct |
6 |
Correct |
24 ms |
5972 KB |
Output is correct |
7 |
Correct |
23 ms |
5972 KB |
Output is correct |
8 |
Correct |
23 ms |
6440 KB |
Output is correct |
9 |
Correct |
26 ms |
5968 KB |
Output is correct |
10 |
Correct |
25 ms |
6276 KB |
Output is correct |
11 |
Correct |
15 ms |
5224 KB |
Output is correct |
12 |
Correct |
15 ms |
5468 KB |
Output is correct |
13 |
Correct |
15 ms |
5192 KB |
Output is correct |
14 |
Correct |
15 ms |
5212 KB |
Output is correct |
15 |
Correct |
24 ms |
6080 KB |
Output is correct |
16 |
Correct |
16 ms |
5200 KB |
Output is correct |
17 |
Correct |
16 ms |
5468 KB |
Output is correct |
18 |
Correct |
24 ms |
6236 KB |
Output is correct |
19 |
Correct |
23 ms |
6096 KB |
Output is correct |
20 |
Correct |
16 ms |
5236 KB |
Output is correct |
21 |
Correct |
26 ms |
6340 KB |
Output is correct |
22 |
Correct |
26 ms |
6232 KB |
Output is correct |
23 |
Correct |
26 ms |
6228 KB |
Output is correct |
24 |
Correct |
26 ms |
6268 KB |
Output is correct |
25 |
Correct |
26 ms |
6436 KB |
Output is correct |
26 |
Correct |
26 ms |
6228 KB |
Output is correct |
27 |
Correct |
35 ms |
6404 KB |
Output is correct |
28 |
Correct |
26 ms |
6232 KB |
Output is correct |
29 |
Correct |
27 ms |
6228 KB |
Output is correct |
30 |
Correct |
26 ms |
6276 KB |
Output is correct |
31 |
Correct |
1 ms |
1112 KB |
Output is correct |
32 |
Correct |
1 ms |
1116 KB |
Output is correct |
33 |
Correct |
1 ms |
1116 KB |
Output is correct |
34 |
Correct |
1 ms |
1116 KB |
Output is correct |
35 |
Correct |
1 ms |
1116 KB |
Output is correct |
36 |
Correct |
1 ms |
1128 KB |
Output is correct |
37 |
Correct |
1 ms |
1116 KB |
Output is correct |
38 |
Correct |
1 ms |
1112 KB |
Output is correct |
39 |
Correct |
1 ms |
1116 KB |
Output is correct |
40 |
Correct |
1 ms |
1116 KB |
Output is correct |
41 |
Correct |
3 ms |
1116 KB |
Output is correct |
42 |
Correct |
3 ms |
1116 KB |
Output is correct |
43 |
Correct |
2 ms |
1116 KB |
Output is correct |
44 |
Correct |
3 ms |
1116 KB |
Output is correct |
45 |
Correct |
2 ms |
1116 KB |
Output is correct |
46 |
Correct |
2 ms |
1116 KB |
Output is correct |
47 |
Correct |
3 ms |
1116 KB |
Output is correct |
48 |
Correct |
1 ms |
1116 KB |
Output is correct |
49 |
Correct |
1 ms |
1116 KB |
Output is correct |
50 |
Correct |
1 ms |
1116 KB |
Output is correct |
51 |
Runtime error |
182 ms |
262144 KB |
Execution killed with signal 9 |
52 |
Halted |
0 ms |
0 KB |
- |