#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define mp make_pair
typedef long long ll;
typedef pair<long long, long long> pll;
typedef pair<int,int> pii;
typedef vector<long long> vl;
typedef vector<int> vi;
int n, k;
vl arr;
ll dp[2010][2010][2];
ll f(int at, int segments, int taken){
if(at >= n)
return 0;
if(segments > k)
return 0;
if(dp[at][segments][taken] != -1)
return dp[at][segments][taken];
ll best = 0;
best = max(best, arr[at] + f(at + 1, segments, 1));
if(taken == 0)
best = max(best, f(at + 1, segments, 0));
if(segments + 1 < k && taken == 1)
best = max(best, f(at + 1, segments + 1, 0));
return dp[at][segments][taken] = best;
}
int main(){
cin>>n>>k;
arr.resize(n);
int cnt = 0;
for(int i = 0; i < n; i++){
cin>>arr[i];
if(arr[i] < 0) cnt++;
}
if(cnt == 0){
ll ans = 0;
for(int i = 0; i < n; i++) ans += arr[i];
cout<<ans<<endl;
}
else if(cnt == 1){
ll left = 0, right = 0, flag = 0, pos = -1;
for(int i = 0; i < n; i++){
if(arr[i] < 0){
pos = i;
flag = 1;
}
else if(flag == 0) left += arr[i];
else right += arr[i];
}
cout<<max(max(left, right), left + right + arr[pos]);
}
else if(k == 1){
ll ans = 0, help = 0;
for(int i = 0; i < n; i++){
help += arr[i];
if(ans < help) ans = help;
if(help < 0) help = 0;
}
cout<<ans<<endl;
}
else{
memset(dp, -1, sizeof dp);
cout<<f(0, 0, 0)<<endl;
}
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
98 ms |
2576 KB |
Output is correct |
2 |
Correct |
109 ms |
2612 KB |
Output is correct |
3 |
Correct |
102 ms |
2644 KB |
Output is correct |
4 |
Correct |
104 ms |
2644 KB |
Output is correct |
5 |
Correct |
104 ms |
2620 KB |
Output is correct |
6 |
Correct |
101 ms |
2636 KB |
Output is correct |
7 |
Correct |
98 ms |
2572 KB |
Output is correct |
8 |
Correct |
102 ms |
2620 KB |
Output is correct |
9 |
Correct |
101 ms |
2580 KB |
Output is correct |
10 |
Correct |
102 ms |
2516 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
56 ms |
2592 KB |
Output is correct |
2 |
Correct |
56 ms |
2644 KB |
Output is correct |
3 |
Correct |
70 ms |
2584 KB |
Output is correct |
4 |
Incorrect |
71 ms |
2516 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
110 ms |
2620 KB |
Output is correct |
2 |
Correct |
109 ms |
2584 KB |
Output is correct |
3 |
Correct |
114 ms |
2616 KB |
Output is correct |
4 |
Correct |
111 ms |
2584 KB |
Output is correct |
5 |
Correct |
114 ms |
2516 KB |
Output is correct |
6 |
Correct |
127 ms |
2644 KB |
Output is correct |
7 |
Correct |
113 ms |
2644 KB |
Output is correct |
8 |
Correct |
108 ms |
2612 KB |
Output is correct |
9 |
Correct |
111 ms |
2644 KB |
Output is correct |
10 |
Correct |
111 ms |
2644 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
26 ms |
63480 KB |
Output is correct |
3 |
Correct |
27 ms |
63516 KB |
Output is correct |
4 |
Correct |
24 ms |
63512 KB |
Output is correct |
5 |
Correct |
24 ms |
63536 KB |
Output is correct |
6 |
Correct |
24 ms |
63444 KB |
Output is correct |
7 |
Correct |
24 ms |
63444 KB |
Output is correct |
8 |
Correct |
24 ms |
63444 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
24 ms |
63488 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
26 ms |
63480 KB |
Output is correct |
3 |
Correct |
27 ms |
63516 KB |
Output is correct |
4 |
Correct |
24 ms |
63512 KB |
Output is correct |
5 |
Correct |
24 ms |
63536 KB |
Output is correct |
6 |
Correct |
24 ms |
63444 KB |
Output is correct |
7 |
Correct |
24 ms |
63444 KB |
Output is correct |
8 |
Correct |
24 ms |
63444 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
24 ms |
63488 KB |
Output is correct |
11 |
Correct |
27 ms |
63464 KB |
Output is correct |
12 |
Correct |
25 ms |
63444 KB |
Output is correct |
13 |
Correct |
26 ms |
63572 KB |
Output is correct |
14 |
Correct |
25 ms |
63516 KB |
Output is correct |
15 |
Correct |
25 ms |
63564 KB |
Output is correct |
16 |
Correct |
25 ms |
63444 KB |
Output is correct |
17 |
Correct |
26 ms |
63448 KB |
Output is correct |
18 |
Correct |
26 ms |
63572 KB |
Output is correct |
19 |
Correct |
26 ms |
63572 KB |
Output is correct |
20 |
Correct |
26 ms |
63568 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
26 ms |
63480 KB |
Output is correct |
3 |
Correct |
27 ms |
63516 KB |
Output is correct |
4 |
Correct |
24 ms |
63512 KB |
Output is correct |
5 |
Correct |
24 ms |
63536 KB |
Output is correct |
6 |
Correct |
24 ms |
63444 KB |
Output is correct |
7 |
Correct |
24 ms |
63444 KB |
Output is correct |
8 |
Correct |
24 ms |
63444 KB |
Output is correct |
9 |
Correct |
0 ms |
212 KB |
Output is correct |
10 |
Correct |
24 ms |
63488 KB |
Output is correct |
11 |
Correct |
27 ms |
63464 KB |
Output is correct |
12 |
Correct |
25 ms |
63444 KB |
Output is correct |
13 |
Correct |
26 ms |
63572 KB |
Output is correct |
14 |
Correct |
25 ms |
63516 KB |
Output is correct |
15 |
Correct |
25 ms |
63564 KB |
Output is correct |
16 |
Correct |
25 ms |
63444 KB |
Output is correct |
17 |
Correct |
26 ms |
63448 KB |
Output is correct |
18 |
Correct |
26 ms |
63572 KB |
Output is correct |
19 |
Correct |
26 ms |
63572 KB |
Output is correct |
20 |
Correct |
26 ms |
63568 KB |
Output is correct |
21 |
Correct |
36 ms |
63700 KB |
Output is correct |
22 |
Correct |
61 ms |
63700 KB |
Output is correct |
23 |
Correct |
49 ms |
63736 KB |
Output is correct |
24 |
Correct |
38 ms |
63700 KB |
Output is correct |
25 |
Correct |
45 ms |
63580 KB |
Output is correct |
26 |
Correct |
35 ms |
63592 KB |
Output is correct |
27 |
Correct |
42 ms |
63696 KB |
Output is correct |
28 |
Correct |
29 ms |
63632 KB |
Output is correct |
29 |
Correct |
27 ms |
63600 KB |
Output is correct |
30 |
Correct |
25 ms |
63672 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
98 ms |
2576 KB |
Output is correct |
2 |
Correct |
109 ms |
2612 KB |
Output is correct |
3 |
Correct |
102 ms |
2644 KB |
Output is correct |
4 |
Correct |
104 ms |
2644 KB |
Output is correct |
5 |
Correct |
104 ms |
2620 KB |
Output is correct |
6 |
Correct |
101 ms |
2636 KB |
Output is correct |
7 |
Correct |
98 ms |
2572 KB |
Output is correct |
8 |
Correct |
102 ms |
2620 KB |
Output is correct |
9 |
Correct |
101 ms |
2580 KB |
Output is correct |
10 |
Correct |
102 ms |
2516 KB |
Output is correct |
11 |
Correct |
56 ms |
2592 KB |
Output is correct |
12 |
Correct |
56 ms |
2644 KB |
Output is correct |
13 |
Correct |
70 ms |
2584 KB |
Output is correct |
14 |
Incorrect |
71 ms |
2516 KB |
Output isn't correct |
15 |
Halted |
0 ms |
0 KB |
- |