// Success consists of going from failure to failure without loss of enthusiasm
#include <bits/stdc++.h>
using namespace std;
#define nl '\n'
#define pb push_back
#define pf push_front
#define mp make_pair
#define f first
#define s second
#define sz(x) int(x.size())
template<class T> using V = vector<T>;
using pi = pair<int, int>;
using vi = V<int>;
using vpi = V<pi>;
using ll = long long;
using pl = pair<ll, ll>;
using vpl = V<pl>;
using vl = V<ll>;
using db = long double;
template<class T> using pq = priority_queue<T, V<T>, greater<T>>;
const int MOD = 1e9 + 7;
const ll INFL = ll(1e17);
const int LG = 20;
int main() {
cin.tie(0)->sync_with_stdio(0);
int N, S; ll K; cin >> N >> S >> K;
vl A(N); for(int i = 1; i < N; i++) {
cin >> A[i]; A[i] += A[i-1];
}
vl X(N); for(auto& x : X) cin >> x;
vl P = {0}; for(auto& x : X) P.pb(P.back() + x);
auto query = [&](int l, int r) {
return P[r + 1] - P[l];
};
vpi nxt(N+1);
for(int i = 0; i < N; i++) {
int j = (--upper_bound(begin(A), end(A), A[i] + K)) - begin(A);
int k = (--upper_bound(begin(A), end(A), A[j] + K)) - begin(A);
nxt[i] = mp(k + 1, j); // first not chosen is k + 1, placed the shield on j
}
nxt[N] = mp(N, -1);
V<vi> up(N+1, vi(LG));
for(int i = N; i >= 0; i--) {
up[i][0] = nxt[i].f;
for(int x = 1; x < LG; x++) up[i][x] = up[up[i][x-1]][x-1];
}
auto jmp = [&](int u, int d) {
for(int i = 0; i < LG; i++) if ((d >> i) & 1) u = up[u][i];
return u;
};
ll ans = -1, best = -1;
for(int l = 0; l < N; l++) {
int r = jmp(l, S) - 1;
ll res = query(l, r);
if (ans < res) ans = res, best = l;
}
vi pos;
while(S--) {
auto [nbest, loc] = nxt[best];
if (loc != -1) pos.pb(loc + 1);
best = nbest;
}
cout << sz(pos) << nl;
for(auto& x : pos) cout << x << " ";
cout << nl;
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1620 KB |
Output is correct |
3 |
Correct |
4 ms |
1748 KB |
Output is correct |
4 |
Correct |
2 ms |
852 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
3 ms |
1492 KB |
Output is correct |
7 |
Correct |
3 ms |
1620 KB |
Output is correct |
8 |
Correct |
3 ms |
1748 KB |
Output is correct |
9 |
Correct |
3 ms |
1236 KB |
Output is correct |
10 |
Correct |
4 ms |
1748 KB |
Output is correct |
11 |
Correct |
4 ms |
1748 KB |
Output is correct |
12 |
Correct |
2 ms |
852 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
253 ms |
127864 KB |
Output is correct |
2 |
Correct |
178 ms |
81364 KB |
Output is correct |
3 |
Correct |
207 ms |
87196 KB |
Output is correct |
4 |
Correct |
249 ms |
95004 KB |
Output is correct |
5 |
Correct |
316 ms |
111440 KB |
Output is correct |
6 |
Correct |
462 ms |
148536 KB |
Output is correct |
7 |
Correct |
283 ms |
98148 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1620 KB |
Output is correct |
3 |
Correct |
4 ms |
1748 KB |
Output is correct |
4 |
Correct |
2 ms |
852 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
3 ms |
1492 KB |
Output is correct |
7 |
Correct |
3 ms |
1620 KB |
Output is correct |
8 |
Correct |
3 ms |
1748 KB |
Output is correct |
9 |
Correct |
3 ms |
1236 KB |
Output is correct |
10 |
Correct |
4 ms |
1748 KB |
Output is correct |
11 |
Correct |
4 ms |
1748 KB |
Output is correct |
12 |
Correct |
2 ms |
852 KB |
Output is correct |
13 |
Correct |
253 ms |
127864 KB |
Output is correct |
14 |
Correct |
178 ms |
81364 KB |
Output is correct |
15 |
Correct |
207 ms |
87196 KB |
Output is correct |
16 |
Correct |
249 ms |
95004 KB |
Output is correct |
17 |
Correct |
316 ms |
111440 KB |
Output is correct |
18 |
Correct |
462 ms |
148536 KB |
Output is correct |
19 |
Correct |
283 ms |
98148 KB |
Output is correct |
20 |
Correct |
185 ms |
81136 KB |
Output is correct |
21 |
Correct |
292 ms |
108516 KB |
Output is correct |
22 |
Correct |
358 ms |
128840 KB |
Output is correct |
23 |
Correct |
256 ms |
130308 KB |
Output is correct |
24 |
Correct |
236 ms |
119988 KB |
Output is correct |
25 |
Correct |
317 ms |
119880 KB |
Output is correct |
26 |
Correct |
201 ms |
89264 KB |
Output is correct |
27 |
Correct |
290 ms |
114720 KB |
Output is correct |
28 |
Correct |
296 ms |
113508 KB |
Output is correct |
29 |
Correct |
369 ms |
147376 KB |
Output is correct |
30 |
Correct |
352 ms |
120696 KB |
Output is correct |
31 |
Correct |
307 ms |
105904 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
448 ms |
129664 KB |
Output is correct |
2 |
Correct |
213 ms |
75660 KB |
Output is correct |
3 |
Correct |
231 ms |
81116 KB |
Output is correct |
4 |
Correct |
337 ms |
117392 KB |
Output is correct |
5 |
Correct |
329 ms |
113856 KB |
Output is correct |
6 |
Correct |
346 ms |
120624 KB |
Output is correct |
7 |
Correct |
539 ms |
156740 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1620 KB |
Output is correct |
3 |
Correct |
4 ms |
1748 KB |
Output is correct |
4 |
Correct |
2 ms |
852 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
3 ms |
1492 KB |
Output is correct |
7 |
Correct |
3 ms |
1620 KB |
Output is correct |
8 |
Correct |
3 ms |
1748 KB |
Output is correct |
9 |
Correct |
3 ms |
1236 KB |
Output is correct |
10 |
Correct |
4 ms |
1748 KB |
Output is correct |
11 |
Correct |
4 ms |
1748 KB |
Output is correct |
12 |
Correct |
2 ms |
852 KB |
Output is correct |
13 |
Correct |
3 ms |
1748 KB |
Output is correct |
14 |
Correct |
3 ms |
1620 KB |
Output is correct |
15 |
Correct |
3 ms |
1492 KB |
Output is correct |
16 |
Correct |
3 ms |
1108 KB |
Output is correct |
17 |
Correct |
3 ms |
1492 KB |
Output is correct |
18 |
Correct |
6 ms |
1872 KB |
Output is correct |
19 |
Correct |
3 ms |
1492 KB |
Output is correct |
20 |
Correct |
4 ms |
1364 KB |
Output is correct |
21 |
Correct |
5 ms |
2072 KB |
Output is correct |
22 |
Correct |
4 ms |
1864 KB |
Output is correct |
23 |
Correct |
4 ms |
1876 KB |
Output is correct |
24 |
Correct |
5 ms |
1620 KB |
Output is correct |
25 |
Correct |
6 ms |
1748 KB |
Output is correct |
26 |
Correct |
4 ms |
1748 KB |
Output is correct |
27 |
Correct |
3 ms |
1364 KB |
Output is correct |
28 |
Correct |
3 ms |
1492 KB |
Output is correct |
29 |
Correct |
3 ms |
1480 KB |
Output is correct |
30 |
Correct |
3 ms |
1748 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1620 KB |
Output is correct |
3 |
Correct |
4 ms |
1748 KB |
Output is correct |
4 |
Correct |
2 ms |
852 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
3 ms |
1492 KB |
Output is correct |
7 |
Correct |
3 ms |
1620 KB |
Output is correct |
8 |
Correct |
3 ms |
1748 KB |
Output is correct |
9 |
Correct |
3 ms |
1236 KB |
Output is correct |
10 |
Correct |
4 ms |
1748 KB |
Output is correct |
11 |
Correct |
4 ms |
1748 KB |
Output is correct |
12 |
Correct |
2 ms |
852 KB |
Output is correct |
13 |
Correct |
253 ms |
127864 KB |
Output is correct |
14 |
Correct |
178 ms |
81364 KB |
Output is correct |
15 |
Correct |
207 ms |
87196 KB |
Output is correct |
16 |
Correct |
249 ms |
95004 KB |
Output is correct |
17 |
Correct |
316 ms |
111440 KB |
Output is correct |
18 |
Correct |
462 ms |
148536 KB |
Output is correct |
19 |
Correct |
283 ms |
98148 KB |
Output is correct |
20 |
Correct |
185 ms |
81136 KB |
Output is correct |
21 |
Correct |
292 ms |
108516 KB |
Output is correct |
22 |
Correct |
358 ms |
128840 KB |
Output is correct |
23 |
Correct |
256 ms |
130308 KB |
Output is correct |
24 |
Correct |
236 ms |
119988 KB |
Output is correct |
25 |
Correct |
317 ms |
119880 KB |
Output is correct |
26 |
Correct |
201 ms |
89264 KB |
Output is correct |
27 |
Correct |
290 ms |
114720 KB |
Output is correct |
28 |
Correct |
296 ms |
113508 KB |
Output is correct |
29 |
Correct |
369 ms |
147376 KB |
Output is correct |
30 |
Correct |
352 ms |
120696 KB |
Output is correct |
31 |
Correct |
307 ms |
105904 KB |
Output is correct |
32 |
Correct |
301 ms |
141592 KB |
Output is correct |
33 |
Correct |
260 ms |
108380 KB |
Output is correct |
34 |
Correct |
410 ms |
149064 KB |
Output is correct |
35 |
Correct |
239 ms |
89784 KB |
Output is correct |
36 |
Correct |
245 ms |
96028 KB |
Output is correct |
37 |
Correct |
274 ms |
106032 KB |
Output is correct |
38 |
Correct |
492 ms |
161196 KB |
Output is correct |
39 |
Correct |
313 ms |
121072 KB |
Output is correct |
40 |
Correct |
458 ms |
162472 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
212 KB |
Output is correct |
2 |
Correct |
4 ms |
1620 KB |
Output is correct |
3 |
Correct |
4 ms |
1748 KB |
Output is correct |
4 |
Correct |
2 ms |
852 KB |
Output is correct |
5 |
Correct |
0 ms |
212 KB |
Output is correct |
6 |
Correct |
3 ms |
1492 KB |
Output is correct |
7 |
Correct |
3 ms |
1620 KB |
Output is correct |
8 |
Correct |
3 ms |
1748 KB |
Output is correct |
9 |
Correct |
3 ms |
1236 KB |
Output is correct |
10 |
Correct |
4 ms |
1748 KB |
Output is correct |
11 |
Correct |
4 ms |
1748 KB |
Output is correct |
12 |
Correct |
2 ms |
852 KB |
Output is correct |
13 |
Correct |
253 ms |
127864 KB |
Output is correct |
14 |
Correct |
178 ms |
81364 KB |
Output is correct |
15 |
Correct |
207 ms |
87196 KB |
Output is correct |
16 |
Correct |
249 ms |
95004 KB |
Output is correct |
17 |
Correct |
316 ms |
111440 KB |
Output is correct |
18 |
Correct |
462 ms |
148536 KB |
Output is correct |
19 |
Correct |
283 ms |
98148 KB |
Output is correct |
20 |
Correct |
185 ms |
81136 KB |
Output is correct |
21 |
Correct |
292 ms |
108516 KB |
Output is correct |
22 |
Correct |
358 ms |
128840 KB |
Output is correct |
23 |
Correct |
256 ms |
130308 KB |
Output is correct |
24 |
Correct |
236 ms |
119988 KB |
Output is correct |
25 |
Correct |
317 ms |
119880 KB |
Output is correct |
26 |
Correct |
201 ms |
89264 KB |
Output is correct |
27 |
Correct |
290 ms |
114720 KB |
Output is correct |
28 |
Correct |
296 ms |
113508 KB |
Output is correct |
29 |
Correct |
369 ms |
147376 KB |
Output is correct |
30 |
Correct |
352 ms |
120696 KB |
Output is correct |
31 |
Correct |
307 ms |
105904 KB |
Output is correct |
32 |
Correct |
448 ms |
129664 KB |
Output is correct |
33 |
Correct |
213 ms |
75660 KB |
Output is correct |
34 |
Correct |
231 ms |
81116 KB |
Output is correct |
35 |
Correct |
337 ms |
117392 KB |
Output is correct |
36 |
Correct |
329 ms |
113856 KB |
Output is correct |
37 |
Correct |
346 ms |
120624 KB |
Output is correct |
38 |
Correct |
539 ms |
156740 KB |
Output is correct |
39 |
Correct |
3 ms |
1748 KB |
Output is correct |
40 |
Correct |
3 ms |
1620 KB |
Output is correct |
41 |
Correct |
3 ms |
1492 KB |
Output is correct |
42 |
Correct |
3 ms |
1108 KB |
Output is correct |
43 |
Correct |
3 ms |
1492 KB |
Output is correct |
44 |
Correct |
6 ms |
1872 KB |
Output is correct |
45 |
Correct |
3 ms |
1492 KB |
Output is correct |
46 |
Correct |
4 ms |
1364 KB |
Output is correct |
47 |
Correct |
5 ms |
2072 KB |
Output is correct |
48 |
Correct |
4 ms |
1864 KB |
Output is correct |
49 |
Correct |
4 ms |
1876 KB |
Output is correct |
50 |
Correct |
5 ms |
1620 KB |
Output is correct |
51 |
Correct |
6 ms |
1748 KB |
Output is correct |
52 |
Correct |
4 ms |
1748 KB |
Output is correct |
53 |
Correct |
3 ms |
1364 KB |
Output is correct |
54 |
Correct |
3 ms |
1492 KB |
Output is correct |
55 |
Correct |
3 ms |
1480 KB |
Output is correct |
56 |
Correct |
3 ms |
1748 KB |
Output is correct |
57 |
Correct |
301 ms |
141592 KB |
Output is correct |
58 |
Correct |
260 ms |
108380 KB |
Output is correct |
59 |
Correct |
410 ms |
149064 KB |
Output is correct |
60 |
Correct |
239 ms |
89784 KB |
Output is correct |
61 |
Correct |
245 ms |
96028 KB |
Output is correct |
62 |
Correct |
274 ms |
106032 KB |
Output is correct |
63 |
Correct |
492 ms |
161196 KB |
Output is correct |
64 |
Correct |
313 ms |
121072 KB |
Output is correct |
65 |
Correct |
458 ms |
162472 KB |
Output is correct |
66 |
Correct |
211 ms |
97172 KB |
Output is correct |
67 |
Correct |
406 ms |
139616 KB |
Output is correct |
68 |
Correct |
287 ms |
98740 KB |
Output is correct |
69 |
Correct |
372 ms |
127604 KB |
Output is correct |
70 |
Correct |
223 ms |
100988 KB |
Output is correct |
71 |
Correct |
509 ms |
161356 KB |
Output is correct |
72 |
Correct |
262 ms |
101812 KB |
Output is correct |
73 |
Correct |
340 ms |
133344 KB |
Output is correct |
74 |
Correct |
244 ms |
83004 KB |
Output is correct |
75 |
Correct |
407 ms |
136860 KB |
Output is correct |
76 |
Correct |
250 ms |
95272 KB |
Output is correct |
77 |
Correct |
358 ms |
127640 KB |
Output is correct |
78 |
Correct |
237 ms |
90812 KB |
Output is correct |
79 |
Correct |
324 ms |
117824 KB |
Output is correct |
80 |
Correct |
433 ms |
147332 KB |
Output is correct |
81 |
Correct |
202 ms |
97072 KB |
Output is correct |
82 |
Correct |
312 ms |
122912 KB |
Output is correct |
83 |
Correct |
209 ms |
81796 KB |
Output is correct |
84 |
Correct |
219 ms |
88796 KB |
Output is correct |
85 |
Correct |
210 ms |
83024 KB |
Output is correct |
86 |
Correct |
356 ms |
146760 KB |
Output is correct |
87 |
Correct |
207 ms |
83388 KB |
Output is correct |
88 |
Correct |
437 ms |
147008 KB |
Output is correct |
89 |
Correct |
321 ms |
118924 KB |
Output is correct |
90 |
Correct |
281 ms |
107716 KB |
Output is correct |
91 |
Correct |
521 ms |
177364 KB |
Output is correct |
92 |
Correct |
469 ms |
163672 KB |
Output is correct |
93 |
Correct |
564 ms |
170276 KB |
Output is correct |
94 |
Correct |
668 ms |
170304 KB |
Output is correct |
95 |
Correct |
419 ms |
144728 KB |
Output is correct |
96 |
Correct |
343 ms |
113776 KB |
Output is correct |
97 |
Correct |
445 ms |
155248 KB |
Output is correct |
98 |
Correct |
335 ms |
103580 KB |
Output is correct |
99 |
Correct |
338 ms |
120296 KB |
Output is correct |
100 |
Correct |
346 ms |
116552 KB |
Output is correct |