#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N=1000006;
int n,s,d;
int a[N],v[N],pref[N],p[N],gr[N],par[N];
vector<int> b,g[N];
deque<int> q;
void DFS(int v,int pr){
q.push_back(v);
int pos=max(0LL,(int)q.size()-(int)s-1LL);
gr[v]=max(q[pos],0LL);
par[v]=pr;
for(auto to : g[v]){
if(pr==to)
continue;
DFS(to,v);
}
q.pop_back();
}
main(){
cin>>n>>s>>d;
b.push_back(0);
for(int k=2;k<=n;k++){
cin>>a[k];
p[k]=p[k-1]+a[k];
b.push_back(p[k]);
}
for(int k=1;k<=n;k++){
cin>>v[k];
pref[k]=pref[k-1]+v[k];
}
for(int k=1;k<=n;k++){
int l=lower_bound(b.begin(),b.end(),p[k]-d)-b.begin();
int ll=lower_bound(b.begin(),b.end(),p[l+1]-d)-b.begin();
g[ll].push_back(k);
}
DFS(0,0);
int st=0,mx=0;
for(int k=1;k<=n;k++){;
if(mx<pref[k]-pref[gr[k]]){
mx=pref[k]-pref[gr[k]];
st=k;
}
}
vector<int> ans;
int m=s;
while(m){
int l=lower_bound(b.begin(),b.end(),p[st]-d)-b.begin()+1;
ans.push_back(l);
st=par[st];
m--;
}
cout<<ans.size()<<endl;
sort(ans.begin(),ans.end());
for(auto i : ans)
cout<<i<<' ';
return 0;
}
Compilation message
SolarStorm.cpp:21:6: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
21 | main(){
| ^
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
23788 KB |
Output is correct |
2 |
Correct |
24 ms |
25452 KB |
Output is correct |
3 |
Correct |
25 ms |
25580 KB |
Output is correct |
4 |
Correct |
20 ms |
24556 KB |
Output is correct |
5 |
Correct |
16 ms |
23788 KB |
Output is correct |
6 |
Correct |
22 ms |
24556 KB |
Output is correct |
7 |
Correct |
22 ms |
24556 KB |
Output is correct |
8 |
Correct |
22 ms |
24704 KB |
Output is correct |
9 |
Correct |
21 ms |
24428 KB |
Output is correct |
10 |
Correct |
23 ms |
24812 KB |
Output is correct |
11 |
Correct |
24 ms |
24684 KB |
Output is correct |
12 |
Correct |
19 ms |
24556 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
532 ms |
85200 KB |
Output is correct |
2 |
Correct |
357 ms |
66120 KB |
Output is correct |
3 |
Correct |
410 ms |
75464 KB |
Output is correct |
4 |
Correct |
526 ms |
84296 KB |
Output is correct |
5 |
Correct |
621 ms |
94664 KB |
Output is correct |
6 |
Correct |
745 ms |
118856 KB |
Output is correct |
7 |
Correct |
488 ms |
102092 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
23788 KB |
Output is correct |
2 |
Correct |
24 ms |
25452 KB |
Output is correct |
3 |
Correct |
25 ms |
25580 KB |
Output is correct |
4 |
Correct |
20 ms |
24556 KB |
Output is correct |
5 |
Correct |
16 ms |
23788 KB |
Output is correct |
6 |
Correct |
22 ms |
24556 KB |
Output is correct |
7 |
Correct |
22 ms |
24556 KB |
Output is correct |
8 |
Correct |
22 ms |
24704 KB |
Output is correct |
9 |
Correct |
21 ms |
24428 KB |
Output is correct |
10 |
Correct |
23 ms |
24812 KB |
Output is correct |
11 |
Correct |
24 ms |
24684 KB |
Output is correct |
12 |
Correct |
19 ms |
24556 KB |
Output is correct |
13 |
Correct |
532 ms |
85200 KB |
Output is correct |
14 |
Correct |
357 ms |
66120 KB |
Output is correct |
15 |
Correct |
410 ms |
75464 KB |
Output is correct |
16 |
Correct |
526 ms |
84296 KB |
Output is correct |
17 |
Correct |
621 ms |
94664 KB |
Output is correct |
18 |
Correct |
745 ms |
118856 KB |
Output is correct |
19 |
Correct |
488 ms |
102092 KB |
Output is correct |
20 |
Correct |
436 ms |
65484 KB |
Output is correct |
21 |
Correct |
618 ms |
137544 KB |
Output is correct |
22 |
Correct |
730 ms |
159048 KB |
Output is correct |
23 |
Correct |
562 ms |
83784 KB |
Output is correct |
24 |
Correct |
575 ms |
80196 KB |
Output is correct |
25 |
Correct |
643 ms |
86984 KB |
Output is correct |
26 |
Correct |
463 ms |
68684 KB |
Output is correct |
27 |
Correct |
620 ms |
84808 KB |
Output is correct |
28 |
Correct |
635 ms |
85324 KB |
Output is correct |
29 |
Correct |
829 ms |
103892 KB |
Output is correct |
30 |
Correct |
671 ms |
104400 KB |
Output is correct |
31 |
Correct |
597 ms |
113100 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
732 ms |
169160 KB |
Output is correct |
2 |
Correct |
390 ms |
107616 KB |
Output is correct |
3 |
Correct |
414 ms |
113992 KB |
Output is correct |
4 |
Correct |
594 ms |
154312 KB |
Output is correct |
5 |
Correct |
574 ms |
150472 KB |
Output is correct |
6 |
Correct |
608 ms |
157896 KB |
Output is correct |
7 |
Correct |
853 ms |
197320 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
23788 KB |
Output is correct |
2 |
Correct |
24 ms |
25452 KB |
Output is correct |
3 |
Correct |
25 ms |
25580 KB |
Output is correct |
4 |
Correct |
20 ms |
24556 KB |
Output is correct |
5 |
Correct |
16 ms |
23788 KB |
Output is correct |
6 |
Correct |
22 ms |
24556 KB |
Output is correct |
7 |
Correct |
22 ms |
24556 KB |
Output is correct |
8 |
Correct |
22 ms |
24704 KB |
Output is correct |
9 |
Correct |
21 ms |
24428 KB |
Output is correct |
10 |
Correct |
23 ms |
24812 KB |
Output is correct |
11 |
Correct |
24 ms |
24684 KB |
Output is correct |
12 |
Correct |
19 ms |
24556 KB |
Output is correct |
13 |
Incorrect |
25 ms |
24812 KB |
Output isn't correct |
14 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
23788 KB |
Output is correct |
2 |
Correct |
24 ms |
25452 KB |
Output is correct |
3 |
Correct |
25 ms |
25580 KB |
Output is correct |
4 |
Correct |
20 ms |
24556 KB |
Output is correct |
5 |
Correct |
16 ms |
23788 KB |
Output is correct |
6 |
Correct |
22 ms |
24556 KB |
Output is correct |
7 |
Correct |
22 ms |
24556 KB |
Output is correct |
8 |
Correct |
22 ms |
24704 KB |
Output is correct |
9 |
Correct |
21 ms |
24428 KB |
Output is correct |
10 |
Correct |
23 ms |
24812 KB |
Output is correct |
11 |
Correct |
24 ms |
24684 KB |
Output is correct |
12 |
Correct |
19 ms |
24556 KB |
Output is correct |
13 |
Correct |
532 ms |
85200 KB |
Output is correct |
14 |
Correct |
357 ms |
66120 KB |
Output is correct |
15 |
Correct |
410 ms |
75464 KB |
Output is correct |
16 |
Correct |
526 ms |
84296 KB |
Output is correct |
17 |
Correct |
621 ms |
94664 KB |
Output is correct |
18 |
Correct |
745 ms |
118856 KB |
Output is correct |
19 |
Correct |
488 ms |
102092 KB |
Output is correct |
20 |
Correct |
436 ms |
65484 KB |
Output is correct |
21 |
Correct |
618 ms |
137544 KB |
Output is correct |
22 |
Correct |
730 ms |
159048 KB |
Output is correct |
23 |
Correct |
562 ms |
83784 KB |
Output is correct |
24 |
Correct |
575 ms |
80196 KB |
Output is correct |
25 |
Correct |
643 ms |
86984 KB |
Output is correct |
26 |
Correct |
463 ms |
68684 KB |
Output is correct |
27 |
Correct |
620 ms |
84808 KB |
Output is correct |
28 |
Correct |
635 ms |
85324 KB |
Output is correct |
29 |
Correct |
829 ms |
103892 KB |
Output is correct |
30 |
Correct |
671 ms |
104400 KB |
Output is correct |
31 |
Correct |
597 ms |
113100 KB |
Output is correct |
32 |
Correct |
679 ms |
90316 KB |
Output is correct |
33 |
Correct |
564 ms |
79232 KB |
Output is correct |
34 |
Correct |
836 ms |
104520 KB |
Output is correct |
35 |
Correct |
506 ms |
72508 KB |
Output is correct |
36 |
Correct |
547 ms |
76240 KB |
Output is correct |
37 |
Correct |
603 ms |
83144 KB |
Output is correct |
38 |
Correct |
890 ms |
141128 KB |
Output is correct |
39 |
Correct |
679 ms |
149320 KB |
Output is correct |
40 |
Correct |
910 ms |
194248 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
16 ms |
23788 KB |
Output is correct |
2 |
Correct |
24 ms |
25452 KB |
Output is correct |
3 |
Correct |
25 ms |
25580 KB |
Output is correct |
4 |
Correct |
20 ms |
24556 KB |
Output is correct |
5 |
Correct |
16 ms |
23788 KB |
Output is correct |
6 |
Correct |
22 ms |
24556 KB |
Output is correct |
7 |
Correct |
22 ms |
24556 KB |
Output is correct |
8 |
Correct |
22 ms |
24704 KB |
Output is correct |
9 |
Correct |
21 ms |
24428 KB |
Output is correct |
10 |
Correct |
23 ms |
24812 KB |
Output is correct |
11 |
Correct |
24 ms |
24684 KB |
Output is correct |
12 |
Correct |
19 ms |
24556 KB |
Output is correct |
13 |
Correct |
532 ms |
85200 KB |
Output is correct |
14 |
Correct |
357 ms |
66120 KB |
Output is correct |
15 |
Correct |
410 ms |
75464 KB |
Output is correct |
16 |
Correct |
526 ms |
84296 KB |
Output is correct |
17 |
Correct |
621 ms |
94664 KB |
Output is correct |
18 |
Correct |
745 ms |
118856 KB |
Output is correct |
19 |
Correct |
488 ms |
102092 KB |
Output is correct |
20 |
Correct |
436 ms |
65484 KB |
Output is correct |
21 |
Correct |
618 ms |
137544 KB |
Output is correct |
22 |
Correct |
730 ms |
159048 KB |
Output is correct |
23 |
Correct |
562 ms |
83784 KB |
Output is correct |
24 |
Correct |
575 ms |
80196 KB |
Output is correct |
25 |
Correct |
643 ms |
86984 KB |
Output is correct |
26 |
Correct |
463 ms |
68684 KB |
Output is correct |
27 |
Correct |
620 ms |
84808 KB |
Output is correct |
28 |
Correct |
635 ms |
85324 KB |
Output is correct |
29 |
Correct |
829 ms |
103892 KB |
Output is correct |
30 |
Correct |
671 ms |
104400 KB |
Output is correct |
31 |
Correct |
597 ms |
113100 KB |
Output is correct |
32 |
Correct |
732 ms |
169160 KB |
Output is correct |
33 |
Correct |
390 ms |
107616 KB |
Output is correct |
34 |
Correct |
414 ms |
113992 KB |
Output is correct |
35 |
Correct |
594 ms |
154312 KB |
Output is correct |
36 |
Correct |
574 ms |
150472 KB |
Output is correct |
37 |
Correct |
608 ms |
157896 KB |
Output is correct |
38 |
Correct |
853 ms |
197320 KB |
Output is correct |
39 |
Incorrect |
25 ms |
24812 KB |
Output isn't correct |
40 |
Halted |
0 ms |
0 KB |
- |