oj.uz

Submission #209425

# Submission time Handle Problem Language Result Execution time Memory
209425 2020-03-14T07:33:08 Z model_code Solar Storm (NOI20_solarstorm) C++17
100 / 100
 482 ms 136412 KB 
#include <bits/stdc++.h>
using namespace std;
constexpr int MAXN=1000000;
long long d[MAXN+1], v[MAXN+1];
int next_k[MAXN];
vector<int> children[MAXN+1];
int ans[MAXN];
int n;
unsigned s;
void dfs(int index, vector<int>& ancestors){
    if(ancestors.size()<s)ans[index]=n;
    else ans[index]=ancestors[ancestors.size()-s];
    ancestors.push_back(index);
    for(const int x : children[index]){
        dfs(x, ancestors);
    }
    ancestors.pop_back();
}
int main(){
    ios_base::sync_with_stdio(false);
    long long k;
    cin>>n>>s>>k;
    assert(n<=MAXN);
    d[0]=0;
    for(int i=1;i<n;++i){
        int tmp;
        cin>>tmp;
        d[i]=d[i-1]+tmp;
    }
    d[n]=numeric_limits<long long>::max();
    v[0]=0;
    for(int i=0;i<n;++i){
        int tmp;
        cin>>tmp;
        v[i+1]=v[i]+tmp;
    }
    int j=0;
    for(int i=0;i<n;++i){
        while(d[j+1]-d[i]<=k)++j;
        next_k[i]=j;
    }
    for(int i=0;i<n;++i){
        children[next_k[next_k[i]]+1].push_back(i);
    }
    vector<int> ancestors;
    for(const int x : children[n]) dfs(x, ancestors);
    int best_index=0;
    for(int i=1;i<n;++i){
        // change '>' to '>=' and the answer should also be correct
        if(v[ans[i]]-v[i]>v[ans[best_index]]-v[best_index])best_index=i;
    }
    unsigned counter=0;
    for(int tmp=best_index;tmp!=n&&counter!=s;tmp=next_k[next_k[tmp]]+1,++counter);
    cout<<counter<<'\n';
    best_index=next_k[best_index];
    cout<<best_index+1;
    while(--counter>0){
        best_index=next_k[next_k[best_index]+1];
        cout<<' '<<best_index+1;
    }
    cout<<'\n';
}

Subtask #1
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 23800 KB Output is correct
2 Correct 27 ms 24828 KB Output is correct
3 Correct 23 ms 24952 KB Output is correct
4 Correct 21 ms 24312 KB Output is correct
5 Correct 18 ms 23928 KB Output is correct
6 Correct 20 ms 24032 KB Output is correct
7 Correct 25 ms 24184 KB Output is correct
8 Correct 22 ms 24184 KB Output is correct
9 Correct 25 ms 24184 KB Output is correct
10 Correct 25 ms 24312 KB Output is correct
11 Correct 24 ms 24312 KB Output is correct
12 Correct 24 ms 24312 KB Output is correct

Subtask #2
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 231 ms 47432 KB Output is correct
2 Correct 161 ms 41548 KB Output is correct
3 Correct 164 ms 51564 KB Output is correct
4 Correct 203 ms 58744 KB Output is correct
5 Correct 274 ms 64760 KB Output is correct
6 Correct 315 ms 78712 KB Output is correct
7 Correct 211 ms 71148 KB Output is correct

Subtask #3
11.0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 23800 KB Output is correct
2 Correct 27 ms 24828 KB Output is correct
3 Correct 23 ms 24952 KB Output is correct
4 Correct 21 ms 24312 KB Output is correct
5 Correct 18 ms 23928 KB Output is correct
6 Correct 20 ms 24032 KB Output is correct
7 Correct 25 ms 24184 KB Output is correct
8 Correct 22 ms 24184 KB Output is correct
9 Correct 25 ms 24184 KB Output is correct
10 Correct 25 ms 24312 KB Output is correct
11 Correct 24 ms 24312 KB Output is correct
12 Correct 24 ms 24312 KB Output is correct
13 Correct 231 ms 47432 KB Output is correct
14 Correct 161 ms 41548 KB Output is correct
15 Correct 164 ms 51564 KB Output is correct
16 Correct 203 ms 58744 KB Output is correct
17 Correct 274 ms 64760 KB Output is correct
18 Correct 315 ms 78712 KB Output is correct
19 Correct 211 ms 71148 KB Output is correct
20 Correct 180 ms 38856 KB Output is correct
21 Correct 259 ms 94408 KB Output is correct
22 Correct 311 ms 107616 KB Output is correct
23 Correct 213 ms 46340 KB Output is correct
24 Correct 230 ms 44132 KB Output is correct
25 Correct 269 ms 52436 KB Output is correct
26 Correct 181 ms 42600 KB Output is correct
27 Correct 249 ms 51952 KB Output is correct
28 Correct 269 ms 52728 KB Output is correct
29 Correct 470 ms 61560 KB Output is correct
30 Correct 294 ms 65260 KB Output is correct
31 Correct 254 ms 74576 KB Output is correct

Subtask #4
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 324 ms 112476 KB Output is correct
2 Correct 182 ms 77388 KB Output is correct
3 Correct 301 ms 82380 KB Output is correct
4 Correct 258 ms 107100 KB Output is correct
5 Correct 247 ms 104664 KB Output is correct
6 Correct 267 ms 109372 KB Output is correct
7 Correct 355 ms 133340 KB Output is correct

Subtask #5
18.0 / 18.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 23800 KB Output is correct
2 Correct 27 ms 24828 KB Output is correct
3 Correct 23 ms 24952 KB Output is correct
4 Correct 21 ms 24312 KB Output is correct
5 Correct 18 ms 23928 KB Output is correct
6 Correct 20 ms 24032 KB Output is correct
7 Correct 25 ms 24184 KB Output is correct
8 Correct 22 ms 24184 KB Output is correct
9 Correct 25 ms 24184 KB Output is correct
10 Correct 25 ms 24312 KB Output is correct
11 Correct 24 ms 24312 KB Output is correct
12 Correct 24 ms 24312 KB Output is correct
13 Correct 26 ms 24184 KB Output is correct
14 Correct 22 ms 24184 KB Output is correct
15 Correct 23 ms 24188 KB Output is correct
16 Correct 21 ms 24184 KB Output is correct
17 Correct 22 ms 24572 KB Output is correct
18 Correct 22 ms 24952 KB Output is correct
19 Correct 27 ms 24696 KB Output is correct
20 Correct 22 ms 24572 KB Output is correct
21 Correct 24 ms 24952 KB Output is correct
22 Correct 23 ms 24952 KB Output is correct
23 Correct 23 ms 24184 KB Output is correct
24 Correct 24 ms 24248 KB Output is correct
25 Correct 24 ms 24444 KB Output is correct
26 Correct 23 ms 24696 KB Output is correct
27 Correct 22 ms 24568 KB Output is correct
28 Correct 22 ms 24696 KB Output is correct
29 Correct 22 ms 24696 KB Output is correct
30 Correct 23 ms 24184 KB Output is correct

Subtask #6
16.0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 23800 KB Output is correct
2 Correct 27 ms 24828 KB Output is correct
3 Correct 23 ms 24952 KB Output is correct
4 Correct 21 ms 24312 KB Output is correct
5 Correct 18 ms 23928 KB Output is correct
6 Correct 20 ms 24032 KB Output is correct
7 Correct 25 ms 24184 KB Output is correct
8 Correct 22 ms 24184 KB Output is correct
9 Correct 25 ms 24184 KB Output is correct
10 Correct 25 ms 24312 KB Output is correct
11 Correct 24 ms 24312 KB Output is correct
12 Correct 24 ms 24312 KB Output is correct
13 Correct 231 ms 47432 KB Output is correct
14 Correct 161 ms 41548 KB Output is correct
15 Correct 164 ms 51564 KB Output is correct
16 Correct 203 ms 58744 KB Output is correct
17 Correct 274 ms 64760 KB Output is correct
18 Correct 315 ms 78712 KB Output is correct
19 Correct 211 ms 71148 KB Output is correct
20 Correct 180 ms 38856 KB Output is correct
21 Correct 259 ms 94408 KB Output is correct
22 Correct 311 ms 107616 KB Output is correct
23 Correct 213 ms 46340 KB Output is correct
24 Correct 230 ms 44132 KB Output is correct
25 Correct 269 ms 52436 KB Output is correct
26 Correct 181 ms 42600 KB Output is correct
27 Correct 249 ms 51952 KB Output is correct
28 Correct 269 ms 52728 KB Output is correct
29 Correct 470 ms 61560 KB Output is correct
30 Correct 294 ms 65260 KB Output is correct
31 Correct 254 ms 74576 KB Output is correct
32 Correct 259 ms 47840 KB Output is correct
33 Correct 231 ms 47716 KB Output is correct
34 Correct 350 ms 61944 KB Output is correct
35 Correct 209 ms 46712 KB Output is correct
36 Correct 236 ms 48632 KB Output is correct
37 Correct 270 ms 52472 KB Output is correct
38 Correct 384 ms 85352 KB Output is correct
39 Correct 296 ms 101472 KB Output is correct
40 Correct 377 ms 129632 KB Output is correct

Subtask #7
30.0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 23800 KB Output is correct
2 Correct 27 ms 24828 KB Output is correct
3 Correct 23 ms 24952 KB Output is correct
4 Correct 21 ms 24312 KB Output is correct
5 Correct 18 ms 23928 KB Output is correct
6 Correct 20 ms 24032 KB Output is correct
7 Correct 25 ms 24184 KB Output is correct
8 Correct 22 ms 24184 KB Output is correct
9 Correct 25 ms 24184 KB Output is correct
10 Correct 25 ms 24312 KB Output is correct
11 Correct 24 ms 24312 KB Output is correct
12 Correct 24 ms 24312 KB Output is correct
13 Correct 231 ms 47432 KB Output is correct
14 Correct 161 ms 41548 KB Output is correct
15 Correct 164 ms 51564 KB Output is correct
16 Correct 203 ms 58744 KB Output is correct
17 Correct 274 ms 64760 KB Output is correct
18 Correct 315 ms 78712 KB Output is correct
19 Correct 211 ms 71148 KB Output is correct
20 Correct 180 ms 38856 KB Output is correct
21 Correct 259 ms 94408 KB Output is correct
22 Correct 311 ms 107616 KB Output is correct
23 Correct 213 ms 46340 KB Output is correct
24 Correct 230 ms 44132 KB Output is correct
25 Correct 269 ms 52436 KB Output is correct
26 Correct 181 ms 42600 KB Output is correct
27 Correct 249 ms 51952 KB Output is correct
28 Correct 269 ms 52728 KB Output is correct
29 Correct 470 ms 61560 KB Output is correct
30 Correct 294 ms 65260 KB Output is correct
31 Correct 254 ms 74576 KB Output is correct
32 Correct 324 ms 112476 KB Output is correct
33 Correct 182 ms 77388 KB Output is correct
34 Correct 301 ms 82380 KB Output is correct
35 Correct 258 ms 107100 KB Output is correct
36 Correct 247 ms 104664 KB Output is correct
37 Correct 267 ms 109372 KB Output is correct
38 Correct 355 ms 133340 KB Output is correct
39 Correct 26 ms 24184 KB Output is correct
40 Correct 22 ms 24184 KB Output is correct
41 Correct 23 ms 24188 KB Output is correct
42 Correct 21 ms 24184 KB Output is correct
43 Correct 22 ms 24572 KB Output is correct
44 Correct 22 ms 24952 KB Output is correct
45 Correct 27 ms 24696 KB Output is correct
46 Correct 22 ms 24572 KB Output is correct
47 Correct 24 ms 24952 KB Output is correct
48 Correct 23 ms 24952 KB Output is correct
49 Correct 23 ms 24184 KB Output is correct
50 Correct 24 ms 24248 KB Output is correct
51 Correct 24 ms 24444 KB Output is correct
52 Correct 23 ms 24696 KB Output is correct
53 Correct 22 ms 24568 KB Output is correct
54 Correct 22 ms 24696 KB Output is correct
55 Correct 22 ms 24696 KB Output is correct
56 Correct 23 ms 24184 KB Output is correct
57 Correct 259 ms 47840 KB Output is correct
58 Correct 231 ms 47716 KB Output is correct
59 Correct 350 ms 61944 KB Output is correct
60 Correct 209 ms 46712 KB Output is correct
61 Correct 236 ms 48632 KB Output is correct
62 Correct 270 ms 52472 KB Output is correct
63 Correct 384 ms 85352 KB Output is correct
64 Correct 296 ms 101472 KB Output is correct
65 Correct 377 ms 129632 KB Output is correct
66 Correct 186 ms 40392 KB Output is correct
67 Correct 354 ms 113244 KB Output is correct
68 Correct 255 ms 85340 KB Output is correct
69 Correct 327 ms 106284 KB Output is correct
70 Correct 197 ms 41056 KB Output is correct
71 Correct 367 ms 65016 KB Output is correct
72 Correct 256 ms 50040 KB Output is correct
73 Correct 332 ms 59408 KB Output is correct
74 Correct 205 ms 61796 KB Output is correct
75 Correct 323 ms 110048 KB Output is correct
76 Correct 231 ms 85728 KB Output is correct
77 Correct 297 ms 55388 KB Output is correct
78 Correct 222 ms 83176 KB Output is correct
79 Correct 277 ms 100548 KB Output is correct
80 Correct 340 ms 119744 KB Output is correct
81 Correct 248 ms 40288 KB Output is correct
82 Correct 250 ms 51172 KB Output is correct
83 Correct 192 ms 44640 KB Output is correct
84 Correct 219 ms 46456 KB Output is correct
85 Correct 211 ms 45176 KB Output is correct
86 Correct 348 ms 63480 KB Output is correct
87 Correct 203 ms 57964 KB Output is correct
88 Correct 352 ms 61308 KB Output is correct
89 Correct 272 ms 100552 KB Output is correct
90 Correct 263 ms 93408 KB Output is correct
91 Correct 482 ms 136412 KB Output is correct
92 Correct 389 ms 130140 KB Output is correct
93 Correct 422 ms 133340 KB Output is correct
94 Correct 414 ms 133340 KB Output is correct
95 Correct 341 ms 60792 KB Output is correct
96 Correct 285 ms 52984 KB Output is correct
97 Correct 365 ms 68592 KB Output is correct
98 Correct 265 ms 60912 KB Output is correct
99 Correct 312 ms 97500 KB Output is correct
100 Correct 300 ms 96732 KB Output is correct