#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<int> a;
vector<ll> pf;
pair<ll, int> solve(ll cost){
vector<pair<ll, int>> dp(a.size());
pair<ll, int> maxval(0, 0);
for (int i=0; i<a.size(); i++){
dp[i] = max(i ? dp[i-1] : make_pair(0LL, 0),
make_pair(pf[i] - cost + maxval.first, maxval.second + 1));
maxval = max(maxval, make_pair(-pf[i] + dp[i].first, dp[i].second));
}
return dp.back();
}
signed main(){
cin.tie(0)->sync_with_stdio(0);
int n, k; cin >> n >> k;
a.resize(n);
for (int i=0; i<n; i++) cin >> a[i];
pf.resize(n);
for (ll i=0, p=0; i<n; i++){
p = pf[i] = p + a[i];
}
ll bl = 0, br = 1e15;
while (bl < br){
ll bm = (bl + br)>>1;
pair<ll, int> r = solve(bm);
if (r.second <= k) br = bm;
else bl = bm+1;
}
pair<ll, int> r = solve(bl);
cout << r.first + bl*r.second << "\n";
}
Compilation message
feast.cpp: In function 'std::pair<long long int, int> solve(ll)':
feast.cpp:9:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
9 | for (int i=0; i<a.size(); i++){
| ~^~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
127 ms |
11020 KB |
Output is correct |
2 |
Correct |
137 ms |
11188 KB |
Output is correct |
3 |
Correct |
139 ms |
11380 KB |
Output is correct |
4 |
Correct |
134 ms |
11340 KB |
Output is correct |
5 |
Correct |
130 ms |
11192 KB |
Output is correct |
6 |
Correct |
130 ms |
11088 KB |
Output is correct |
7 |
Correct |
148 ms |
11096 KB |
Output is correct |
8 |
Correct |
134 ms |
11276 KB |
Output is correct |
9 |
Correct |
135 ms |
11108 KB |
Output is correct |
10 |
Correct |
131 ms |
11192 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
135 ms |
9388 KB |
Output is correct |
2 |
Correct |
123 ms |
9612 KB |
Output is correct |
3 |
Correct |
118 ms |
9428 KB |
Output is correct |
4 |
Correct |
126 ms |
9516 KB |
Output is correct |
5 |
Correct |
154 ms |
11096 KB |
Output is correct |
6 |
Correct |
120 ms |
9520 KB |
Output is correct |
7 |
Correct |
124 ms |
9592 KB |
Output is correct |
8 |
Correct |
146 ms |
11312 KB |
Output is correct |
9 |
Correct |
136 ms |
11048 KB |
Output is correct |
10 |
Correct |
126 ms |
9612 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
136 ms |
11364 KB |
Output is correct |
2 |
Correct |
140 ms |
11320 KB |
Output is correct |
3 |
Correct |
134 ms |
11440 KB |
Output is correct |
4 |
Correct |
140 ms |
11320 KB |
Output is correct |
5 |
Correct |
139 ms |
11340 KB |
Output is correct |
6 |
Correct |
133 ms |
11448 KB |
Output is correct |
7 |
Correct |
144 ms |
11504 KB |
Output is correct |
8 |
Correct |
140 ms |
11324 KB |
Output is correct |
9 |
Correct |
138 ms |
11564 KB |
Output is correct |
10 |
Correct |
154 ms |
11520 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
320 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
324 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
324 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
320 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
324 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
324 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
212 KB |
Output is correct |
11 |
Correct |
1 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
320 KB |
Output is correct |
15 |
Correct |
1 ms |
212 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
17 |
Correct |
1 ms |
212 KB |
Output is correct |
18 |
Correct |
1 ms |
212 KB |
Output is correct |
19 |
Correct |
1 ms |
212 KB |
Output is correct |
20 |
Correct |
1 ms |
212 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
320 KB |
Output is correct |
2 |
Correct |
0 ms |
212 KB |
Output is correct |
3 |
Correct |
1 ms |
324 KB |
Output is correct |
4 |
Correct |
1 ms |
212 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
1 ms |
324 KB |
Output is correct |
7 |
Correct |
0 ms |
212 KB |
Output is correct |
8 |
Correct |
1 ms |
212 KB |
Output is correct |
9 |
Correct |
1 ms |
212 KB |
Output is correct |
10 |
Correct |
1 ms |
212 KB |
Output is correct |
11 |
Correct |
1 ms |
212 KB |
Output is correct |
12 |
Correct |
1 ms |
212 KB |
Output is correct |
13 |
Correct |
1 ms |
212 KB |
Output is correct |
14 |
Correct |
1 ms |
320 KB |
Output is correct |
15 |
Correct |
1 ms |
212 KB |
Output is correct |
16 |
Correct |
1 ms |
212 KB |
Output is correct |
17 |
Correct |
1 ms |
212 KB |
Output is correct |
18 |
Correct |
1 ms |
212 KB |
Output is correct |
19 |
Correct |
1 ms |
212 KB |
Output is correct |
20 |
Correct |
1 ms |
212 KB |
Output is correct |
21 |
Correct |
2 ms |
340 KB |
Output is correct |
22 |
Correct |
1 ms |
340 KB |
Output is correct |
23 |
Correct |
1 ms |
340 KB |
Output is correct |
24 |
Correct |
1 ms |
388 KB |
Output is correct |
25 |
Correct |
2 ms |
340 KB |
Output is correct |
26 |
Correct |
1 ms |
340 KB |
Output is correct |
27 |
Correct |
1 ms |
340 KB |
Output is correct |
28 |
Correct |
1 ms |
332 KB |
Output is correct |
29 |
Correct |
2 ms |
328 KB |
Output is correct |
30 |
Correct |
2 ms |
340 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
127 ms |
11020 KB |
Output is correct |
2 |
Correct |
137 ms |
11188 KB |
Output is correct |
3 |
Correct |
139 ms |
11380 KB |
Output is correct |
4 |
Correct |
134 ms |
11340 KB |
Output is correct |
5 |
Correct |
130 ms |
11192 KB |
Output is correct |
6 |
Correct |
130 ms |
11088 KB |
Output is correct |
7 |
Correct |
148 ms |
11096 KB |
Output is correct |
8 |
Correct |
134 ms |
11276 KB |
Output is correct |
9 |
Correct |
135 ms |
11108 KB |
Output is correct |
10 |
Correct |
131 ms |
11192 KB |
Output is correct |
11 |
Correct |
135 ms |
9388 KB |
Output is correct |
12 |
Correct |
123 ms |
9612 KB |
Output is correct |
13 |
Correct |
118 ms |
9428 KB |
Output is correct |
14 |
Correct |
126 ms |
9516 KB |
Output is correct |
15 |
Correct |
154 ms |
11096 KB |
Output is correct |
16 |
Correct |
120 ms |
9520 KB |
Output is correct |
17 |
Correct |
124 ms |
9592 KB |
Output is correct |
18 |
Correct |
146 ms |
11312 KB |
Output is correct |
19 |
Correct |
136 ms |
11048 KB |
Output is correct |
20 |
Correct |
126 ms |
9612 KB |
Output is correct |
21 |
Correct |
136 ms |
11364 KB |
Output is correct |
22 |
Correct |
140 ms |
11320 KB |
Output is correct |
23 |
Correct |
134 ms |
11440 KB |
Output is correct |
24 |
Correct |
140 ms |
11320 KB |
Output is correct |
25 |
Correct |
139 ms |
11340 KB |
Output is correct |
26 |
Correct |
133 ms |
11448 KB |
Output is correct |
27 |
Correct |
144 ms |
11504 KB |
Output is correct |
28 |
Correct |
140 ms |
11324 KB |
Output is correct |
29 |
Correct |
138 ms |
11564 KB |
Output is correct |
30 |
Correct |
154 ms |
11520 KB |
Output is correct |
31 |
Correct |
0 ms |
320 KB |
Output is correct |
32 |
Correct |
0 ms |
212 KB |
Output is correct |
33 |
Correct |
1 ms |
324 KB |
Output is correct |
34 |
Correct |
1 ms |
212 KB |
Output is correct |
35 |
Correct |
0 ms |
212 KB |
Output is correct |
36 |
Correct |
1 ms |
324 KB |
Output is correct |
37 |
Correct |
0 ms |
212 KB |
Output is correct |
38 |
Correct |
1 ms |
212 KB |
Output is correct |
39 |
Correct |
1 ms |
212 KB |
Output is correct |
40 |
Correct |
1 ms |
212 KB |
Output is correct |
41 |
Correct |
1 ms |
212 KB |
Output is correct |
42 |
Correct |
1 ms |
212 KB |
Output is correct |
43 |
Correct |
1 ms |
212 KB |
Output is correct |
44 |
Correct |
1 ms |
320 KB |
Output is correct |
45 |
Correct |
1 ms |
212 KB |
Output is correct |
46 |
Correct |
1 ms |
212 KB |
Output is correct |
47 |
Correct |
1 ms |
212 KB |
Output is correct |
48 |
Correct |
1 ms |
212 KB |
Output is correct |
49 |
Correct |
1 ms |
212 KB |
Output is correct |
50 |
Correct |
1 ms |
212 KB |
Output is correct |
51 |
Correct |
2 ms |
340 KB |
Output is correct |
52 |
Correct |
1 ms |
340 KB |
Output is correct |
53 |
Correct |
1 ms |
340 KB |
Output is correct |
54 |
Correct |
1 ms |
388 KB |
Output is correct |
55 |
Correct |
2 ms |
340 KB |
Output is correct |
56 |
Correct |
1 ms |
340 KB |
Output is correct |
57 |
Correct |
1 ms |
340 KB |
Output is correct |
58 |
Correct |
1 ms |
332 KB |
Output is correct |
59 |
Correct |
2 ms |
328 KB |
Output is correct |
60 |
Correct |
2 ms |
340 KB |
Output is correct |
61 |
Correct |
141 ms |
11300 KB |
Output is correct |
62 |
Correct |
152 ms |
11596 KB |
Output is correct |
63 |
Correct |
137 ms |
11264 KB |
Output is correct |
64 |
Correct |
158 ms |
11516 KB |
Output is correct |
65 |
Correct |
149 ms |
11468 KB |
Output is correct |
66 |
Correct |
152 ms |
11460 KB |
Output is correct |
67 |
Correct |
149 ms |
11300 KB |
Output is correct |
68 |
Correct |
140 ms |
11504 KB |
Output is correct |
69 |
Correct |
136 ms |
11284 KB |
Output is correct |
70 |
Correct |
136 ms |
11244 KB |
Output is correct |