oj.uz

Submission #956468

# Submission time Handle Problem Language Result Execution time Memory
956468 2024-04-02T03:14:12 Z kilkuwu A Difficult(y) Choice (BOI21_books) C++17
100 / 100
 1 ms 1444 KB 
#include <bits/stdc++.h>

#include "books.h"

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.
//

void solve(int N, int K, long long A, int S) {
  // first K - 1 books
  // and first one >= A 
  // >= 2 * A -> impossible
  std::vector<long long> mem(N, -1);

  auto ask = [&](int i) {
    assert(0 <= i && i < N);
    if (mem[i] != -1) return mem[i];
    return mem[i] = skim(i + 1);
  };

  long long sum = 0;
  for (int i = 0; i < K; i++) {
    sum += ask(i);
  }

  if (sum > 2 * A) {
    impossible();
    return;
  }
  
  if (sum >= A) {
    std::vector<int> res(K);
    std::iota(res.begin(), res.end(), 1);
    answer(res);
    return;
  }

  // sum < A now
  int lb = K, rb = N;
  while (lb < rb) {
    int mid = (lb + rb) / 2;
    if (ask(mid) > A) {
      rb = mid;
    } else {
      lb = mid + 1;
    }
  }

  int id = rb;

  if (id < N) {
    auto ssum = sum + ask(id) - ask(K - 1);
    if (ssum >= A && ssum <= 2 * A) {
      std::vector<int> res;
      for (int i = 0; i < K - 1; i++) {
        res.emplace_back(i + 1);
      }
      res.emplace_back(id + 1);
      answer(res);
      return;
    }
    
  }

  std::vector<int> res_right;
  for (int i = id - 1; i >= std::max(K, id - K); i--) {
    res_right.emplace_back(i);
    ask(i);
  }

  // we know first sol is not working

  std::vector<int> res_left;
  for (int i = 0; i < K; i++) {
    res_left.push_back(i);
  }

  while (res_right.size()) {
    sum -= ask(res_left.front());
    res_left.erase(res_left.begin());
    sum += ask(res_right.back());
    res_left.emplace_back(res_right.back());
    res_right.pop_back();

    if (sum >= A) {
      // we good 
      for (int& x : res_left) x++;
      answer(res_left);
      return;
    }
  }

  impossible();
}

Subtask #1
5.0 / 5.0

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

Subtask #2
15.0 / 15.0

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

Subtask #3
10.0 / 10.0

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

Subtask #4
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 1112 KB Output is correct
2 Correct 1 ms 1368 KB Output is correct
3 Correct 1 ms 1112 KB Output is correct
4 Correct 1 ms 1368 KB Output is correct
5 Correct 1 ms 1112 KB Output is correct
6 Correct 1 ms 1112 KB Output is correct
7 Correct 1 ms 1112 KB Output is correct
8 Correct 1 ms 1368 KB Output is correct
9 Correct 1 ms 1112 KB Output is correct
10 Correct 1 ms 1368 KB Output is correct
11 Correct 1 ms 1112 KB Output is correct
12 Correct 1 ms 1368 KB Output is correct
13 Correct 1 ms 1112 KB Output is correct
14 Correct 1 ms 1112 KB Output is correct
15 Correct 1 ms 1196 KB Output is correct
16 Correct 1 ms 1112 KB Output is correct
17 Correct 1 ms 1112 KB Output is correct
18 Correct 1 ms 1112 KB Output is correct
19 Correct 1 ms 1112 KB Output is correct
20 Correct 1 ms 1112 KB Output is correct
21 Correct 1 ms 1372 KB Output is correct
22 Correct 1 ms 1112 KB Output is correct
23 Correct 1 ms 1108 KB Output is correct
24 Correct 1 ms 1368 KB Output is correct
25 Correct 1 ms 1112 KB Output is correct

Subtask #5
15.0 / 15.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 1112 KB Output is correct
2 Correct 1 ms 1368 KB Output is correct
3 Correct 1 ms 1112 KB Output is correct
4 Correct 1 ms 1368 KB Output is correct
5 Correct 1 ms 1112 KB Output is correct
6 Correct 1 ms 1112 KB Output is correct
7 Correct 1 ms 1112 KB Output is correct
8 Correct 1 ms 1368 KB Output is correct
9 Correct 1 ms 1112 KB Output is correct
10 Correct 1 ms 1368 KB Output is correct
11 Correct 1 ms 1112 KB Output is correct
12 Correct 1 ms 1368 KB Output is correct
13 Correct 1 ms 1112 KB Output is correct
14 Correct 1 ms 1112 KB Output is correct
15 Correct 1 ms 1196 KB Output is correct
16 Correct 1 ms 1112 KB Output is correct
17 Correct 1 ms 1112 KB Output is correct
18 Correct 1 ms 1112 KB Output is correct
19 Correct 1 ms 1112 KB Output is correct
20 Correct 1 ms 1112 KB Output is correct
21 Correct 1 ms 1372 KB Output is correct
22 Correct 1 ms 1112 KB Output is correct
23 Correct 1 ms 1108 KB Output is correct
24 Correct 1 ms 1368 KB Output is correct
25 Correct 1 ms 1112 KB Output is correct
26 Correct 1 ms 1112 KB Output is correct
27 Correct 1 ms 1264 KB Output is correct
28 Correct 1 ms 1272 KB Output is correct
29 Correct 1 ms 1112 KB Output is correct
30 Correct 1 ms 1196 KB Output is correct
31 Correct 1 ms 1112 KB Output is correct
32 Correct 1 ms 1112 KB Output is correct
33 Correct 1 ms 1200 KB Output is correct
34 Correct 1 ms 1112 KB Output is correct
35 Correct 1 ms 1112 KB Output is correct
36 Correct 1 ms 1112 KB Output is correct
37 Correct 1 ms 1112 KB Output is correct
38 Correct 1 ms 1112 KB Output is correct
39 Correct 1 ms 1112 KB Output is correct
40 Correct 1 ms 1364 KB Output is correct
41 Correct 1 ms 1368 KB Output is correct
42 Correct 1 ms 1112 KB Output is correct
43 Correct 1 ms 1204 KB Output is correct
44 Correct 1 ms 1112 KB Output is correct
45 Correct 1 ms 1196 KB Output is correct
46 Correct 1 ms 1112 KB Output is correct
47 Correct 1 ms 1196 KB Output is correct
48 Correct 1 ms 1112 KB Output is correct
49 Correct 1 ms 1112 KB Output is correct
50 Correct 1 ms 1372 KB Output is correct

Subtask #6
20.0 / 20.0

# Verdict Execution time Memory Grader output
1 Correct 1 ms 1112 KB Output is correct
2 Correct 1 ms 1368 KB Output is correct
3 Correct 1 ms 1112 KB Output is correct
4 Correct 1 ms 1368 KB Output is correct
5 Correct 1 ms 1112 KB Output is correct
6 Correct 1 ms 1112 KB Output is correct
7 Correct 1 ms 1112 KB Output is correct
8 Correct 1 ms 1368 KB Output is correct
9 Correct 1 ms 1112 KB Output is correct
10 Correct 1 ms 1368 KB Output is correct
11 Correct 1 ms 1112 KB Output is correct
12 Correct 1 ms 1368 KB Output is correct
13 Correct 1 ms 1112 KB Output is correct
14 Correct 1 ms 1112 KB Output is correct
15 Correct 1 ms 1196 KB Output is correct
16 Correct 1 ms 1112 KB Output is correct
17 Correct 1 ms 1112 KB Output is correct
18 Correct 1 ms 1112 KB Output is correct
19 Correct 1 ms 1112 KB Output is correct
20 Correct 1 ms 1112 KB Output is correct
21 Correct 1 ms 1372 KB Output is correct
22 Correct 1 ms 1112 KB Output is correct
23 Correct 1 ms 1108 KB Output is correct
24 Correct 1 ms 1368 KB Output is correct
25 Correct 1 ms 1112 KB Output is correct
26 Correct 1 ms 1116 KB Output is correct
27 Correct 1 ms 1372 KB Output is correct
28 Correct 1 ms 1116 KB Output is correct
29 Correct 1 ms 1120 KB Output is correct
30 Correct 1 ms 1208 KB Output is correct
31 Correct 1 ms 1120 KB Output is correct
32 Correct 1 ms 1372 KB Output is correct
33 Correct 1 ms 1188 KB Output is correct
34 Correct 1 ms 1116 KB Output is correct
35 Correct 1 ms 1372 KB Output is correct
36 Correct 1 ms 1116 KB Output is correct
37 Correct 1 ms 1116 KB Output is correct
38 Correct 1 ms 1376 KB Output is correct
39 Correct 1 ms 1120 KB Output is correct
40 Correct 1 ms 1376 KB Output is correct
41 Correct 1 ms 1120 KB Output is correct
42 Correct 1 ms 1112 KB Output is correct
43 Correct 1 ms 1112 KB Output is correct
44 Correct 1 ms 1192 KB Output is correct
45 Correct 1 ms 1112 KB Output is correct
46 Correct 1 ms 1444 KB Output is correct
47 Correct 1 ms 1112 KB Output is correct
48 Correct 1 ms 1368 KB Output is correct
49 Correct 1 ms 1112 KB Output is correct
50 Correct 1 ms 1112 KB Output is correct

Subtask #7
20.0 / 20.0

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