-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathlist_of_books.cpp
79 lines (60 loc) · 1.31 KB
/
list_of_books.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include <bits/stdc++.h>
using namespace std;
#define long long long int
struct FenwickTree {
private:
int n;
vector<int> tree;
public:
FenwickTree(int n) {
this->n = n + 1;
tree.assign(n + 1, 0);
}
void add(int i, int v) {
if (!i) tree[0] += v;
else {
while (i < n) {
tree[i] += v;
i += LSO(i);
}
}
}
int sum(int i, int j) {
return sum(j) - sum(i - 1);
}
private:
int LSO(int i) {
return i & -i;
}
int sum(int i) {
if (i < 0) return 0;
int ans = tree[0];
while (i > 0) {
ans += tree[i];
i -= LSO(i);
}
return ans;
}
};
int main() {
ios_base::sync_with_stdio(false);
cin.tie(NULL);
int M; cin >> M;
vector<int> books(M + 1);
for (int i = 1; i <= M; i++)
cin >> books[i];
FenwickTree tree(M);
int N; cin >> N;
while (N--) {
int idx; cin >> idx;
int l = 1, r = M, aux = 1;
while (l <= r) {
int mid = (l + r) / 2;
if (mid - tree.sum(1, mid) >= idx) r = mid - 1;
else l = mid + 1, aux = mid + 1;
}
int ans = books[aux];
cout << ans << "\n";
tree.add(aux, 1);
}
}