oj.uz

Submission #470306

# Submission time Handle Problem Language Result Execution time Memory
470306 2021-09-03T12:46:26 Z Haruto810198 A Difficult(y) Choice (BOI21_books) C++17
100 / 100
 2 ms 328 KB 
#include <bits/stdc++.h>

#include "books.h"

#define ll long long
#define FOR(i, l, r, d) for(int i=(l); i<=(r); i+=(d))
#define vi vector<int>
#define pb push_back

using namespace std;
//
// --- Sample implementation for the task books ---
//
// To compile this program with the sample grader, place:
//     books.h books_sample.cpp sample_grader.cpp
// in a single folder and run:
//     g++ books_sample.cpp sample_grader.cpp
// in this folder.
//

const int MAX = 100010;

ll arr[MAX];
vector<int> res;

ll query(int id){
    if(arr[id] == 0){
        arr[id] = skim(id);
    }
    return arr[id];
}

bool solve_for_small_items(int small_cnt, int K, ll A){

    vi pref, suf;

    if(small_cnt < K){
        return 0;
    }

    FOR(i, 1, K, 1){
        pref.pb(i);
        suf.pb(small_cnt + 1 - i);
    }

    FOR(i, 0, K, 1){

        ll sum = 0;
        res.clear();

        FOR(j, 0, i-1, 1){
            sum += query(pref[j]);
            res.pb(pref[j]);
        }
        FOR(j, 0, K-i-1, 1){
            sum += query(suf[j]);
            res.pb(suf[j]);
        }

        if(A <= sum and sum <= 2*A){
            return 1;
        }

    }

    return 0;

}

void solve(int N, int K, ll A, int Q){

    // find first large item
    int first_large_item;
    int L = 1, R = N, mid;

    if(query(N) < A){
        first_large_item = N + 1;
    }
    else{
        while(L < R){
            mid = (L + R) / 2;
            if(query(mid) >= A) R = mid;
            else L = mid + 1;
        }
        first_large_item = L;
    }

    // all small items
    if(solve_for_small_items(first_large_item - 1, K, A)){
        answer(res);
        return;
    }

    // if no solution for all small items:

    // 1 large item
    if(first_large_item == N + 1){ // no large item
        impossible();
        return;
    }

    if(first_large_item < K){ // not enough small items
        impossible();
        return;
    }

    ll sum = 0;
    res.clear();

    FOR(i, 1, K-1, 1){ // solution for 1 large item and K-1 small items
        sum += query(i);
        res.pb(i);
    }
    sum += query(first_large_item);
    res.pb(first_large_item);

    if(A <= sum and sum <= 2*A){
        answer(res);
    }
    else{
        impossible();
    }
    return;

}

Subtask #1
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 0 ms 200 KB Output is correct
3 Correct 1 ms 200 KB Output is correct
4 Correct 1 ms 200 KB Output is correct
5 Correct 1 ms 200 KB Output is correct
6 Correct 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 1 ms 200 KB Output is correct

Subtask #2
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 200 KB Output is correct
3 Correct 0 ms 200 KB Output is correct
4 Correct 0 ms 200 KB Output is correct
5 Correct 0 ms 200 KB Output is correct
6 Correct 1 ms 200 KB Output is correct
7 Correct 1 ms 200 KB Output is correct
8 Correct 1 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 200 KB Output is correct
11 Correct 1 ms 328 KB Output is correct
12 Correct 1 ms 292 KB Output is correct
13 Correct 1 ms 200 KB Output is correct
14 Correct 1 ms 304 KB Output is correct
15 Correct 1 ms 304 KB Output is correct
16 Correct 1 ms 200 KB Output is correct

Subtask #3
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 320 KB Output is correct
3 Correct 1 ms 296 KB Output is correct
4 Correct 1 ms 296 KB Output is correct
5 Correct 1 ms 292 KB Output is correct
6 Correct 1 ms 292 KB Output is correct
7 Correct 1 ms 284 KB Output is correct
8 Correct 1 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct

Subtask #4
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 320 KB Output is correct
3 Correct 1 ms 296 KB Output is correct
4 Correct 1 ms 296 KB Output is correct
5 Correct 1 ms 292 KB Output is correct
6 Correct 1 ms 292 KB Output is correct
7 Correct 1 ms 284 KB Output is correct
8 Correct 1 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 296 KB Output is correct
11 Correct 1 ms 200 KB Output is correct
12 Correct 1 ms 296 KB Output is correct
13 Correct 1 ms 200 KB Output is correct
14 Correct 1 ms 308 KB Output is correct
15 Correct 1 ms 200 KB Output is correct
16 Correct 1 ms 284 KB Output is correct
17 Correct 1 ms 200 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 200 KB Output is correct
20 Correct 1 ms 200 KB Output is correct
21 Correct 1 ms 292 KB Output is correct
22 Correct 1 ms 296 KB Output is correct
23 Correct 1 ms 200 KB Output is correct
24 Correct 1 ms 328 KB Output is correct
25 Correct 1 ms 296 KB Output is correct

Subtask #5
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 320 KB Output is correct
3 Correct 1 ms 296 KB Output is correct
4 Correct 1 ms 296 KB Output is correct
5 Correct 1 ms 292 KB Output is correct
6 Correct 1 ms 292 KB Output is correct
7 Correct 1 ms 284 KB Output is correct
8 Correct 1 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 296 KB Output is correct
11 Correct 1 ms 200 KB Output is correct
12 Correct 1 ms 296 KB Output is correct
13 Correct 1 ms 200 KB Output is correct
14 Correct 1 ms 308 KB Output is correct
15 Correct 1 ms 200 KB Output is correct
16 Correct 1 ms 284 KB Output is correct
17 Correct 1 ms 200 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 200 KB Output is correct
20 Correct 1 ms 200 KB Output is correct
21 Correct 1 ms 292 KB Output is correct
22 Correct 1 ms 296 KB Output is correct
23 Correct 1 ms 200 KB Output is correct
24 Correct 1 ms 328 KB Output is correct
25 Correct 1 ms 296 KB Output is correct
26 Correct 2 ms 284 KB Output is correct
27 Correct 1 ms 288 KB Output is correct
28 Correct 1 ms 320 KB Output is correct
29 Correct 1 ms 200 KB Output is correct
30 Correct 1 ms 300 KB Output is correct
31 Correct 1 ms 280 KB Output is correct
32 Correct 1 ms 200 KB Output is correct
33 Correct 1 ms 200 KB Output is correct
34 Correct 1 ms 200 KB Output is correct
35 Correct 1 ms 200 KB Output is correct
36 Correct 1 ms 284 KB Output is correct
37 Correct 1 ms 296 KB Output is correct
38 Correct 1 ms 200 KB Output is correct
39 Correct 1 ms 328 KB Output is correct
40 Correct 1 ms 200 KB Output is correct
41 Correct 1 ms 304 KB Output is correct
42 Correct 1 ms 200 KB Output is correct
43 Correct 1 ms 200 KB Output is correct
44 Correct 1 ms 200 KB Output is correct
45 Correct 1 ms 320 KB Output is correct
46 Correct 1 ms 280 KB Output is correct
47 Correct 1 ms 320 KB Output is correct
48 Correct 1 ms 328 KB Output is correct
49 Correct 1 ms 328 KB Output is correct
50 Correct 1 ms 200 KB Output is correct

Subtask #6
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 320 KB Output is correct
3 Correct 1 ms 296 KB Output is correct
4 Correct 1 ms 296 KB Output is correct
5 Correct 1 ms 292 KB Output is correct
6 Correct 1 ms 292 KB Output is correct
7 Correct 1 ms 284 KB Output is correct
8 Correct 1 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 296 KB Output is correct
11 Correct 1 ms 200 KB Output is correct
12 Correct 1 ms 296 KB Output is correct
13 Correct 1 ms 200 KB Output is correct
14 Correct 1 ms 308 KB Output is correct
15 Correct 1 ms 200 KB Output is correct
16 Correct 1 ms 284 KB Output is correct
17 Correct 1 ms 200 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 200 KB Output is correct
20 Correct 1 ms 200 KB Output is correct
21 Correct 1 ms 292 KB Output is correct
22 Correct 1 ms 296 KB Output is correct
23 Correct 1 ms 200 KB Output is correct
24 Correct 1 ms 328 KB Output is correct
25 Correct 1 ms 296 KB Output is correct
26 Correct 1 ms 296 KB Output is correct
27 Correct 1 ms 328 KB Output is correct
28 Correct 1 ms 328 KB Output is correct
29 Correct 1 ms 284 KB Output is correct
30 Correct 1 ms 200 KB Output is correct
31 Correct 1 ms 296 KB Output is correct
32 Correct 1 ms 296 KB Output is correct
33 Correct 1 ms 200 KB Output is correct
34 Correct 1 ms 200 KB Output is correct
35 Correct 1 ms 288 KB Output is correct
36 Correct 1 ms 200 KB Output is correct
37 Correct 1 ms 300 KB Output is correct
38 Correct 1 ms 280 KB Output is correct
39 Correct 1 ms 296 KB Output is correct
40 Correct 1 ms 200 KB Output is correct
41 Correct 1 ms 328 KB Output is correct
42 Correct 1 ms 304 KB Output is correct
43 Correct 1 ms 200 KB Output is correct
44 Correct 1 ms 304 KB Output is correct
45 Correct 1 ms 300 KB Output is correct
46 Correct 1 ms 200 KB Output is correct
47 Correct 1 ms 260 KB Output is correct
48 Correct 1 ms 292 KB Output is correct
49 Correct 1 ms 284 KB Output is correct
50 Correct 1 ms 328 KB Output is correct

Subtask #7
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 200 KB Output is correct
2 Correct 1 ms 320 KB Output is correct
3 Correct 1 ms 296 KB Output is correct
4 Correct 1 ms 296 KB Output is correct
5 Correct 1 ms 292 KB Output is correct
6 Correct 1 ms 292 KB Output is correct
7 Correct 1 ms 284 KB Output is correct
8 Correct 1 ms 200 KB Output is correct
9 Correct 1 ms 200 KB Output is correct
10 Correct 1 ms 296 KB Output is correct
11 Correct 1 ms 200 KB Output is correct
12 Correct 1 ms 296 KB Output is correct
13 Correct 1 ms 200 KB Output is correct
14 Correct 1 ms 308 KB Output is correct
15 Correct 1 ms 200 KB Output is correct
16 Correct 1 ms 284 KB Output is correct
17 Correct 1 ms 200 KB Output is correct
18 Correct 1 ms 200 KB Output is correct
19 Correct 1 ms 200 KB Output is correct
20 Correct 1 ms 200 KB Output is correct
21 Correct 1 ms 292 KB Output is correct
22 Correct 1 ms 296 KB Output is correct
23 Correct 1 ms 200 KB Output is correct
24 Correct 1 ms 328 KB Output is correct
25 Correct 1 ms 296 KB Output is correct
26 Correct 2 ms 284 KB Output is correct
27 Correct 1 ms 288 KB Output is correct
28 Correct 1 ms 320 KB Output is correct
29 Correct 1 ms 200 KB Output is correct
30 Correct 1 ms 300 KB Output is correct
31 Correct 1 ms 280 KB Output is correct
32 Correct 1 ms 200 KB Output is correct
33 Correct 1 ms 200 KB Output is correct
34 Correct 1 ms 200 KB Output is correct
35 Correct 1 ms 200 KB Output is correct
36 Correct 1 ms 284 KB Output is correct
37 Correct 1 ms 296 KB Output is correct
38 Correct 1 ms 200 KB Output is correct
39 Correct 1 ms 328 KB Output is correct
40 Correct 1 ms 200 KB Output is correct
41 Correct 1 ms 304 KB Output is correct
42 Correct 1 ms 200 KB Output is correct
43 Correct 1 ms 200 KB Output is correct
44 Correct 1 ms 200 KB Output is correct
45 Correct 1 ms 320 KB Output is correct
46 Correct 1 ms 280 KB Output is correct
47 Correct 1 ms 320 KB Output is correct
48 Correct 1 ms 328 KB Output is correct
49 Correct 1 ms 328 KB Output is correct
50 Correct 1 ms 200 KB Output is correct
51 Correct 1 ms 296 KB Output is correct
52 Correct 1 ms 328 KB Output is correct
53 Correct 1 ms 328 KB Output is correct
54 Correct 1 ms 284 KB Output is correct
55 Correct 1 ms 200 KB Output is correct
56 Correct 1 ms 296 KB Output is correct
57 Correct 1 ms 296 KB Output is correct
58 Correct 1 ms 200 KB Output is correct
59 Correct 1 ms 200 KB Output is correct
60 Correct 1 ms 288 KB Output is correct
61 Correct 1 ms 200 KB Output is correct
62 Correct 1 ms 300 KB Output is correct
63 Correct 1 ms 280 KB Output is correct
64 Correct 1 ms 296 KB Output is correct
65 Correct 1 ms 200 KB Output is correct
66 Correct 1 ms 328 KB Output is correct
67 Correct 1 ms 304 KB Output is correct
68 Correct 1 ms 200 KB Output is correct
69 Correct 1 ms 304 KB Output is correct
70 Correct 1 ms 300 KB Output is correct
71 Correct 1 ms 200 KB Output is correct
72 Correct 1 ms 260 KB Output is correct
73 Correct 1 ms 292 KB Output is correct
74 Correct 1 ms 284 KB Output is correct
75 Correct 1 ms 328 KB Output is correct
76 Correct 1 ms 200 KB Output is correct
77 Correct 1 ms 200 KB Output is correct
78 Correct 1 ms 200 KB Output is correct
79 Correct 0 ms 200 KB Output is correct
80 Correct 1 ms 200 KB Output is correct
81 Correct 1 ms 200 KB Output is correct
82 Correct 1 ms 200 KB Output is correct
83 Correct 1 ms 200 KB Output is correct
84 Correct 1 ms 200 KB Output is correct
85 Correct 1 ms 200 KB Output is correct
86 Correct 1 ms 300 KB Output is correct
87 Correct 1 ms 200 KB Output is correct
88 Correct 1 ms 200 KB Output is correct
89 Correct 1 ms 200 KB Output is correct
90 Correct 1 ms 200 KB Output is correct
91 Correct 1 ms 308 KB Output is correct
92 Correct 1 ms 292 KB Output is correct
93 Correct 1 ms 200 KB Output is correct
94 Correct 1 ms 284 KB Output is correct
95 Correct 1 ms 296 KB Output is correct
96 Correct 1 ms 296 KB Output is correct
97 Correct 1 ms 296 KB Output is correct
98 Correct 1 ms 296 KB Output is correct
99 Correct 1 ms 200 KB Output is correct
100 Correct 1 ms 200 KB Output is correct
101 Correct 1 ms 200 KB Output is correct
102 Correct 1 ms 284 KB Output is correct
103 Correct 1 ms 296 KB Output is correct
104 Correct 1 ms 300 KB Output is correct
105 Correct 1 ms 200 KB Output is correct
106 Correct 1 ms 328 KB Output is correct
107 Correct 1 ms 300 KB Output is correct
108 Correct 1 ms 200 KB Output is correct
109 Correct 1 ms 296 KB Output is correct
110 Correct 1 ms 200 KB Output is correct
111 Correct 1 ms 200 KB Output is correct
112 Correct 1 ms 292 KB Output is correct
113 Correct 1 ms 200 KB Output is correct
114 Correct 1 ms 304 KB Output is correct
115 Correct 1 ms 328 KB Output is correct
116 Correct 1 ms 304 KB Output is correct
117 Correct 1 ms 328 KB Output is correct
118 Correct 1 ms 200 KB Output is correct
119 Correct 1 ms 292 KB Output is correct
120 Correct 1 ms 200 KB Output is correct
121 Correct 1 ms 200 KB Output is correct
122 Correct 1 ms 200 KB Output is correct
123 Correct 1 ms 304 KB Output is correct
124 Correct 1 ms 328 KB Output is correct
125 Correct 1 ms 296 KB Output is correct
126 Correct 1 ms 200 KB Output is correct
127 Correct 1 ms 304 KB Output is correct
128 Correct 1 ms 200 KB Output is correct
129 Correct 1 ms 284 KB Output is correct
130 Correct 1 ms 200 KB Output is correct
131 Correct 1 ms 200 KB Output is correct
132 Correct 1 ms 296 KB Output is correct
133 Correct 1 ms 328 KB Output is correct
134 Correct 1 ms 308 KB Output is correct
135 Correct 1 ms 296 KB Output is correct