#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
const int MOD = 998244353;
ll binpow(ll a, ll p, int mod = MOD) {
ll res = 1;
while (p) {
if (p & 1) {
(res *= a) %= mod;
}
p >>= 1;
(a *= a) %= mod;
}
return res;
}
ll gcd(ll a, ll b) {
return b == 0 ? a : gcd(b, a % b);
}
void solve() {
int s, n;
ll k;
cin >> n >> s >> k;
vector<ll> pos(n + 1);
pos[0] = -1;
for (int i = 2; i <= n; i++) {
cin >> pos[i];
pos[i] += pos[i - 1];
}
vector<ll> value(n + 1);
for (int i = 1; i <= n; i++) {
cin >> value[i];
value[i] += value[i - 1];
}
const int K = 21;
vector<vector<int>> go(n + 2, vector<int>(K));
go[n + 1].assign(K, n + 1);
ll best_value = -1;
ll start = 0;
for (int i = n; i >= 1; i--) {
int mid = upper_bound(pos.begin(), pos.end(), pos[i] + k) - pos.begin() - 1;
int lst = upper_bound(pos.begin(), pos.end(), pos[mid] + k) - pos.begin();
go[i][0] = lst;
for (int j = 1; j < K; j++) {
go[i][j] = go[go[i][j - 1]][j - 1];
}
int cur = i;
for (int j = K - 1; j >= 0; j--) {
if (s >> j & 1) {
cur = go[cur][j];
}
}
ll cur_value = value[cur - 1] - value[i - 1];
if (cur_value > best_value) {
best_value = cur_value;
start = i;
}
}
vector<int> ans;
while (s && start <= n) {
int mid = upper_bound(pos.begin(), pos.end(), pos[start] + k) - pos.begin() - 1;
ans.push_back(mid);
start = go[start][0];
s--;
}
cout << ans.size() << endl;
for (int i : ans) {
cout << i << ' ';
}
cout << endl;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int T = 1;
// cin >> T;
for (int tc = 1; tc <= T; tc++) {
// cout << "Case #" << tc << ": ";
solve();
}
return 0;
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
6 ms |
1604 KB |
Output is correct |
3 |
Correct |
7 ms |
1764 KB |
Output is correct |
4 |
Correct |
4 ms |
844 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Correct |
6 ms |
1356 KB |
Output is correct |
7 |
Correct |
6 ms |
1484 KB |
Output is correct |
8 |
Correct |
6 ms |
1740 KB |
Output is correct |
9 |
Correct |
6 ms |
1204 KB |
Output is correct |
10 |
Correct |
7 ms |
1732 KB |
Output is correct |
11 |
Correct |
7 ms |
1728 KB |
Output is correct |
12 |
Correct |
3 ms |
844 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
451 ms |
121972 KB |
Output is correct |
2 |
Correct |
314 ms |
77528 KB |
Output is correct |
3 |
Correct |
341 ms |
83216 KB |
Output is correct |
4 |
Correct |
435 ms |
90664 KB |
Output is correct |
5 |
Correct |
507 ms |
106180 KB |
Output is correct |
6 |
Correct |
767 ms |
141596 KB |
Output is correct |
7 |
Correct |
575 ms |
93556 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
6 ms |
1604 KB |
Output is correct |
3 |
Correct |
7 ms |
1764 KB |
Output is correct |
4 |
Correct |
4 ms |
844 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Correct |
6 ms |
1356 KB |
Output is correct |
7 |
Correct |
6 ms |
1484 KB |
Output is correct |
8 |
Correct |
6 ms |
1740 KB |
Output is correct |
9 |
Correct |
6 ms |
1204 KB |
Output is correct |
10 |
Correct |
7 ms |
1732 KB |
Output is correct |
11 |
Correct |
7 ms |
1728 KB |
Output is correct |
12 |
Correct |
3 ms |
844 KB |
Output is correct |
13 |
Correct |
451 ms |
121972 KB |
Output is correct |
14 |
Correct |
314 ms |
77528 KB |
Output is correct |
15 |
Correct |
341 ms |
83216 KB |
Output is correct |
16 |
Correct |
435 ms |
90664 KB |
Output is correct |
17 |
Correct |
507 ms |
106180 KB |
Output is correct |
18 |
Correct |
767 ms |
141596 KB |
Output is correct |
19 |
Correct |
575 ms |
93556 KB |
Output is correct |
20 |
Correct |
304 ms |
80020 KB |
Output is correct |
21 |
Correct |
514 ms |
98060 KB |
Output is correct |
22 |
Correct |
615 ms |
116416 KB |
Output is correct |
23 |
Correct |
441 ms |
117544 KB |
Output is correct |
24 |
Correct |
447 ms |
108324 KB |
Output is correct |
25 |
Correct |
450 ms |
108384 KB |
Output is correct |
26 |
Correct |
329 ms |
80756 KB |
Output is correct |
27 |
Correct |
458 ms |
103640 KB |
Output is correct |
28 |
Correct |
492 ms |
102580 KB |
Output is correct |
29 |
Correct |
602 ms |
133280 KB |
Output is correct |
30 |
Correct |
589 ms |
109104 KB |
Output is correct |
31 |
Correct |
502 ms |
95620 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
976 ms |
120160 KB |
Output is correct |
2 |
Correct |
481 ms |
72132 KB |
Output is correct |
3 |
Correct |
418 ms |
77368 KB |
Output is correct |
4 |
Correct |
640 ms |
111968 KB |
Output is correct |
5 |
Correct |
567 ms |
108528 KB |
Output is correct |
6 |
Correct |
607 ms |
114980 KB |
Output is correct |
7 |
Correct |
1984 ms |
147976 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
6 ms |
1604 KB |
Output is correct |
3 |
Correct |
7 ms |
1764 KB |
Output is correct |
4 |
Correct |
4 ms |
844 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Correct |
6 ms |
1356 KB |
Output is correct |
7 |
Correct |
6 ms |
1484 KB |
Output is correct |
8 |
Correct |
6 ms |
1740 KB |
Output is correct |
9 |
Correct |
6 ms |
1204 KB |
Output is correct |
10 |
Correct |
7 ms |
1732 KB |
Output is correct |
11 |
Correct |
7 ms |
1728 KB |
Output is correct |
12 |
Correct |
3 ms |
844 KB |
Output is correct |
13 |
Correct |
7 ms |
1740 KB |
Output is correct |
14 |
Correct |
7 ms |
1484 KB |
Output is correct |
15 |
Correct |
6 ms |
1356 KB |
Output is correct |
16 |
Correct |
5 ms |
972 KB |
Output is correct |
17 |
Correct |
7 ms |
1356 KB |
Output is correct |
18 |
Correct |
7 ms |
1736 KB |
Output is correct |
19 |
Correct |
6 ms |
1484 KB |
Output is correct |
20 |
Correct |
6 ms |
1224 KB |
Output is correct |
21 |
Correct |
8 ms |
1880 KB |
Output is correct |
22 |
Correct |
7 ms |
1732 KB |
Output is correct |
23 |
Correct |
7 ms |
1672 KB |
Output is correct |
24 |
Correct |
6 ms |
1484 KB |
Output is correct |
25 |
Correct |
8 ms |
1612 KB |
Output is correct |
26 |
Correct |
8 ms |
1728 KB |
Output is correct |
27 |
Correct |
6 ms |
1356 KB |
Output is correct |
28 |
Correct |
6 ms |
1356 KB |
Output is correct |
29 |
Correct |
6 ms |
1356 KB |
Output is correct |
30 |
Correct |
7 ms |
1608 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
6 ms |
1604 KB |
Output is correct |
3 |
Correct |
7 ms |
1764 KB |
Output is correct |
4 |
Correct |
4 ms |
844 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Correct |
6 ms |
1356 KB |
Output is correct |
7 |
Correct |
6 ms |
1484 KB |
Output is correct |
8 |
Correct |
6 ms |
1740 KB |
Output is correct |
9 |
Correct |
6 ms |
1204 KB |
Output is correct |
10 |
Correct |
7 ms |
1732 KB |
Output is correct |
11 |
Correct |
7 ms |
1728 KB |
Output is correct |
12 |
Correct |
3 ms |
844 KB |
Output is correct |
13 |
Correct |
451 ms |
121972 KB |
Output is correct |
14 |
Correct |
314 ms |
77528 KB |
Output is correct |
15 |
Correct |
341 ms |
83216 KB |
Output is correct |
16 |
Correct |
435 ms |
90664 KB |
Output is correct |
17 |
Correct |
507 ms |
106180 KB |
Output is correct |
18 |
Correct |
767 ms |
141596 KB |
Output is correct |
19 |
Correct |
575 ms |
93556 KB |
Output is correct |
20 |
Correct |
304 ms |
80020 KB |
Output is correct |
21 |
Correct |
514 ms |
98060 KB |
Output is correct |
22 |
Correct |
615 ms |
116416 KB |
Output is correct |
23 |
Correct |
441 ms |
117544 KB |
Output is correct |
24 |
Correct |
447 ms |
108324 KB |
Output is correct |
25 |
Correct |
450 ms |
108384 KB |
Output is correct |
26 |
Correct |
329 ms |
80756 KB |
Output is correct |
27 |
Correct |
458 ms |
103640 KB |
Output is correct |
28 |
Correct |
492 ms |
102580 KB |
Output is correct |
29 |
Correct |
602 ms |
133280 KB |
Output is correct |
30 |
Correct |
589 ms |
109104 KB |
Output is correct |
31 |
Correct |
502 ms |
95620 KB |
Output is correct |
32 |
Correct |
506 ms |
127960 KB |
Output is correct |
33 |
Correct |
384 ms |
97912 KB |
Output is correct |
34 |
Correct |
665 ms |
134848 KB |
Output is correct |
35 |
Correct |
394 ms |
81188 KB |
Output is correct |
36 |
Correct |
447 ms |
86968 KB |
Output is correct |
37 |
Correct |
440 ms |
95820 KB |
Output is correct |
38 |
Correct |
930 ms |
145804 KB |
Output is correct |
39 |
Correct |
543 ms |
109360 KB |
Output is correct |
40 |
Correct |
856 ms |
146940 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
204 KB |
Output is correct |
2 |
Correct |
6 ms |
1604 KB |
Output is correct |
3 |
Correct |
7 ms |
1764 KB |
Output is correct |
4 |
Correct |
4 ms |
844 KB |
Output is correct |
5 |
Correct |
1 ms |
204 KB |
Output is correct |
6 |
Correct |
6 ms |
1356 KB |
Output is correct |
7 |
Correct |
6 ms |
1484 KB |
Output is correct |
8 |
Correct |
6 ms |
1740 KB |
Output is correct |
9 |
Correct |
6 ms |
1204 KB |
Output is correct |
10 |
Correct |
7 ms |
1732 KB |
Output is correct |
11 |
Correct |
7 ms |
1728 KB |
Output is correct |
12 |
Correct |
3 ms |
844 KB |
Output is correct |
13 |
Correct |
451 ms |
121972 KB |
Output is correct |
14 |
Correct |
314 ms |
77528 KB |
Output is correct |
15 |
Correct |
341 ms |
83216 KB |
Output is correct |
16 |
Correct |
435 ms |
90664 KB |
Output is correct |
17 |
Correct |
507 ms |
106180 KB |
Output is correct |
18 |
Correct |
767 ms |
141596 KB |
Output is correct |
19 |
Correct |
575 ms |
93556 KB |
Output is correct |
20 |
Correct |
304 ms |
80020 KB |
Output is correct |
21 |
Correct |
514 ms |
98060 KB |
Output is correct |
22 |
Correct |
615 ms |
116416 KB |
Output is correct |
23 |
Correct |
441 ms |
117544 KB |
Output is correct |
24 |
Correct |
447 ms |
108324 KB |
Output is correct |
25 |
Correct |
450 ms |
108384 KB |
Output is correct |
26 |
Correct |
329 ms |
80756 KB |
Output is correct |
27 |
Correct |
458 ms |
103640 KB |
Output is correct |
28 |
Correct |
492 ms |
102580 KB |
Output is correct |
29 |
Correct |
602 ms |
133280 KB |
Output is correct |
30 |
Correct |
589 ms |
109104 KB |
Output is correct |
31 |
Correct |
502 ms |
95620 KB |
Output is correct |
32 |
Correct |
976 ms |
120160 KB |
Output is correct |
33 |
Correct |
481 ms |
72132 KB |
Output is correct |
34 |
Correct |
418 ms |
77368 KB |
Output is correct |
35 |
Correct |
640 ms |
111968 KB |
Output is correct |
36 |
Correct |
567 ms |
108528 KB |
Output is correct |
37 |
Correct |
607 ms |
114980 KB |
Output is correct |
38 |
Correct |
1984 ms |
147976 KB |
Output is correct |
39 |
Correct |
7 ms |
1740 KB |
Output is correct |
40 |
Correct |
7 ms |
1484 KB |
Output is correct |
41 |
Correct |
6 ms |
1356 KB |
Output is correct |
42 |
Correct |
5 ms |
972 KB |
Output is correct |
43 |
Correct |
7 ms |
1356 KB |
Output is correct |
44 |
Correct |
7 ms |
1736 KB |
Output is correct |
45 |
Correct |
6 ms |
1484 KB |
Output is correct |
46 |
Correct |
6 ms |
1224 KB |
Output is correct |
47 |
Correct |
8 ms |
1880 KB |
Output is correct |
48 |
Correct |
7 ms |
1732 KB |
Output is correct |
49 |
Correct |
7 ms |
1672 KB |
Output is correct |
50 |
Correct |
6 ms |
1484 KB |
Output is correct |
51 |
Correct |
8 ms |
1612 KB |
Output is correct |
52 |
Correct |
8 ms |
1728 KB |
Output is correct |
53 |
Correct |
6 ms |
1356 KB |
Output is correct |
54 |
Correct |
6 ms |
1356 KB |
Output is correct |
55 |
Correct |
6 ms |
1356 KB |
Output is correct |
56 |
Correct |
7 ms |
1608 KB |
Output is correct |
57 |
Correct |
506 ms |
127960 KB |
Output is correct |
58 |
Correct |
384 ms |
97912 KB |
Output is correct |
59 |
Correct |
665 ms |
134848 KB |
Output is correct |
60 |
Correct |
394 ms |
81188 KB |
Output is correct |
61 |
Correct |
447 ms |
86968 KB |
Output is correct |
62 |
Correct |
440 ms |
95820 KB |
Output is correct |
63 |
Correct |
930 ms |
145804 KB |
Output is correct |
64 |
Correct |
543 ms |
109360 KB |
Output is correct |
65 |
Correct |
856 ms |
146940 KB |
Output is correct |
66 |
Correct |
327 ms |
87820 KB |
Output is correct |
67 |
Correct |
1160 ms |
124816 KB |
Output is correct |
68 |
Correct |
546 ms |
86164 KB |
Output is correct |
69 |
Correct |
1345 ms |
114856 KB |
Output is correct |
70 |
Correct |
358 ms |
91304 KB |
Output is correct |
71 |
Correct |
738 ms |
145804 KB |
Output is correct |
72 |
Correct |
420 ms |
92140 KB |
Output is correct |
73 |
Correct |
530 ms |
120516 KB |
Output is correct |
74 |
Correct |
523 ms |
75016 KB |
Output is correct |
75 |
Correct |
867 ms |
123684 KB |
Output is correct |
76 |
Correct |
460 ms |
86144 KB |
Output is correct |
77 |
Correct |
517 ms |
115408 KB |
Output is correct |
78 |
Correct |
441 ms |
82068 KB |
Output is correct |
79 |
Correct |
655 ms |
106516 KB |
Output is correct |
80 |
Correct |
831 ms |
133196 KB |
Output is correct |
81 |
Correct |
321 ms |
87828 KB |
Output is correct |
82 |
Correct |
454 ms |
111084 KB |
Output is correct |
83 |
Correct |
350 ms |
74088 KB |
Output is correct |
84 |
Correct |
363 ms |
80200 KB |
Output is correct |
85 |
Correct |
341 ms |
75076 KB |
Output is correct |
86 |
Correct |
606 ms |
132704 KB |
Output is correct |
87 |
Correct |
393 ms |
75440 KB |
Output is correct |
88 |
Correct |
705 ms |
132940 KB |
Output is correct |
89 |
Correct |
530 ms |
107480 KB |
Output is correct |
90 |
Correct |
499 ms |
97348 KB |
Output is correct |
91 |
Correct |
967 ms |
157736 KB |
Output is correct |
92 |
Correct |
1330 ms |
147640 KB |
Output is correct |
93 |
Correct |
1768 ms |
152744 KB |
Output is correct |
94 |
Correct |
1861 ms |
152756 KB |
Output is correct |
95 |
Correct |
623 ms |
130920 KB |
Output is correct |
96 |
Correct |
480 ms |
102796 KB |
Output is correct |
97 |
Correct |
769 ms |
139988 KB |
Output is correct |
98 |
Correct |
657 ms |
93020 KB |
Output is correct |
99 |
Correct |
633 ms |
106196 KB |
Output is correct |
100 |
Correct |
629 ms |
102400 KB |
Output is correct |