#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int maxN = 100000, maxU = 200000, INF = 1000000000, BL = 1000;
int a[maxU], b[maxU], h[maxN], n, u, d;
struct comp {
bool operator()(int i, int j) const {
return h[i] == h[j] ? i < j : h[i] < h[j];
}
};
template<typename T>
using uset = set<T, comp>;
vector<uset<int>> g[maxN];
uset<int> temp[maxN];
vector<int> ti[maxN];
vector<pair<int, int>> c[maxN];
void init(int N, int D, int H[]) {
n = N, d = D;
memcpy(h, H, sizeof(h[0]) * n);
}
void curseChanges(int U, int A[], int B[]) {
u = U;
memcpy(a, A, sizeof(a[0]) * u), memcpy(b, B, sizeof(b[0]) * u);
set<pair<int, int>> adj;
for (int i = 0; i < u; ++i) {
if (a[i] > b[i]) {
swap(a[i], b[i]);
}
if (adj.count({a[i], b[i]})) {
adj.erase({a[i], b[i]});
c[a[i]].emplace_back(~b[i], i + 1);
c[b[i]].emplace_back(~a[i], i + 1);
temp[a[i]].erase(b[i]);
temp[b[i]].erase(a[i]);
} else {
adj.insert({a[i], b[i]});
c[a[i]].emplace_back(b[i], i + 1);
c[b[i]].emplace_back(a[i], i + 1);
temp[a[i]].insert(b[i]);
temp[b[i]].insert(a[i]);
}
if (c[a[i]].size() % BL == 0) {
g[a[i]].push_back(temp[a[i]]);
ti[a[i]].push_back(i + 1);
}
if (c[b[i]].size() % BL == 0) {
g[b[i]].push_back(temp[b[i]]);
ti[b[i]].push_back(i + 1);
}
}
}
vector<int> get(int x, int v) {
int it = upper_bound(ti[x].begin(), ti[x].end(), v) - ti[x].begin() - 1, t = 0;
uset<int> ans;
if (it > -1) {
ans = g[x][it];
t = ti[x][it] + 1;
}
it = lower_bound(c[x].begin(), c[x].end(), make_pair(-1, t), [](pair<int, int> a, pair<int, int> b) {
return a.second < b.second;
}) - c[x].begin();
int cnt = 0;
for (; it < int(c[x].size()) && c[x][it].second <= v; ++it) {
auto [y, tt] = c[x][it];
if (y < 0) {
ans.erase(~y);
} else {
ans.insert(y);
}
cnt += 1;
}
assert(cnt <= BL);
vector<int> answer(ans.begin(), ans.end());
return answer;
}
int question(int x, int y, int v) {
int ans = INF;
vector<int> f = get(x, v), s = get(y, v);
int pnt = 0;
for (int to: f) {
while (pnt < s.size() && h[s[pnt]] < h[to]) {
pnt += 1;
}
if (pnt != s.size()) {
ans = min(ans, h[s[pnt]] - h[to]);
}
if (pnt > 0) {
ans = min(ans, h[to] - h[s[pnt - 1]]);
}
}
return ans;
}
Compilation message
potion.cpp: In function 'int question(int, int, int)':
potion.cpp:94:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
94 | while (pnt < s.size() && h[s[pnt]] < h[to]) {
| ~~~~^~~~~~~~~~
potion.cpp:97:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
97 | if (pnt != s.size()) {
| ~~~~^~~~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
11984 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
8 ms |
12240 KB |
Output is correct |
2 |
Correct |
8 ms |
12240 KB |
Output is correct |
3 |
Correct |
8 ms |
12240 KB |
Output is correct |
4 |
Correct |
18 ms |
13012 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
549 ms |
50584 KB |
Output is correct |
2 |
Correct |
559 ms |
50492 KB |
Output is correct |
3 |
Execution timed out |
3080 ms |
19848 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
537 ms |
50528 KB |
Output is correct |
2 |
Execution timed out |
3066 ms |
24104 KB |
Time limit exceeded |
3 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
51 ms |
14032 KB |
Output is correct |
2 |
Correct |
234 ms |
12624 KB |
Output is correct |
3 |
Correct |
2364 ms |
12548 KB |
Output is correct |
4 |
Correct |
2654 ms |
13316 KB |
Output is correct |
5 |
Correct |
2339 ms |
13768 KB |
Output is correct |
6 |
Correct |
167 ms |
13904 KB |
Output is correct |
7 |
Execution timed out |
3091 ms |
12752 KB |
Time limit exceeded |
8 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
6 ms |
11984 KB |
Output is correct |
2 |
Correct |
8 ms |
12240 KB |
Output is correct |
3 |
Correct |
8 ms |
12240 KB |
Output is correct |
4 |
Correct |
8 ms |
12240 KB |
Output is correct |
5 |
Correct |
18 ms |
13012 KB |
Output is correct |
6 |
Correct |
549 ms |
50584 KB |
Output is correct |
7 |
Correct |
559 ms |
50492 KB |
Output is correct |
8 |
Execution timed out |
3080 ms |
19848 KB |
Time limit exceeded |
9 |
Halted |
0 ms |
0 KB |
- |