#ifndef hwe
#include "highway.h"
#endif
#include <bits/stdc++.h>
using namespace std;
#define popcount(n) (__builtin_popcountll((n)))
#define clz(n) (__builtin_clzll((n)))
#define ctz(n) (__builtin_ctzll((n)))
#define lg(n) (63 - __builtin_clzll((n)))
#define BIT(n, i) (((n) >> (i)) & 1ll)
#define MASK(i) (1ll << (i))
#define FLIP(n, i) ((n) ^ (1ll << (i)))
#define ON(n, i) ((n) | MASK(i))
#define OFF(n, i) ((n) & ~MASK(i))
#define Int __int128
#define fi first
#define se second
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<long long, long long> pll;
typedef pair<long long, int> pli;
typedef pair<int, long long> pil;
typedef vector<pair<int, int>> vii;
typedef vector<pair<long long, long long>> vll;
typedef vector<pair<long long, int>> vli;
typedef vector<pair<int, long long>> vil;
template <class T1, class T2>
bool maximize(T1 &x, T2 y) {
if (x < y) {
x = y;
return true;
}
return false;
}
template <class T1, class T2>
bool minimize(T1 &x, T2 y) {
if (x > y) {
x = y;
return true;
}
return false;
}
template <class T>
void remove_duplicate(vector<T> &ve) {
sort (ve.begin(), ve.end());
ve.resize(unique(ve.begin(), ve.end()) - ve.begin());
}
mt19937 rng(chrono::high_resolution_clock::now().time_since_epoch().count());
template <class T> T random(T l, T r) {
return uniform_int_distribution<T>(l, r)(rng);
}
template <class T> T random(T r) {
return rng() % r;
}
const int N = 90000 + 5;
const int MOD = 1e9 + 7;
const int inf = 1e9;
const ll INF = 1e18;
const int LIGHT = 0, HEAVY = 1;
int dep[N], e[N];
vii adj[N];
vector<int> ve[N];
#ifdef hwe
ll ask(const vector<int> &w) {
return 1;
}
void answer(int s, int t) {
cerr << "Answer " << s << ' ' << t << '\n';
}
#endif
void dfs(int u, int fa) {
for (auto [v, id] : adj[u]) if (v != fa) {
dep[v] = dep[u] + 1, e[v] = id;
ve[dep[v]].emplace_back(v);
dfs(v, u);
}
}
void find_pair(int n, vector<int> U, vector<int> V, int A, int B) {
int m = U.size();
for (int i = 0; i < m; ++i) {
adj[U[i]].emplace_back(V[i], i), adj[V[i]].emplace_back(U[i], i);
}
dfs(0, -1);
int low = 1, high = n - 1, depth = 0;
while (low <= high) {
int mid = low + high >> 1; vector<int> w(m, HEAVY); ll expected = 1ll * mid * A;
for (int d = 1; d <= mid; ++d) for (auto v : ve[d]) {
w[e[v]] = LIGHT;
}
if (expected <= ask(w)) low = (depth = mid) + 1;
else high = mid - 1;
}
low = 0, high = ve[depth].size() - 1; int pos = -1; ll expected = 1ll * depth * B;
while (low <= high) {
int mid = low + high >> 1; vector<int> w(m, HEAVY);
for (int i = 0; i < mid; ++i) w[e[ve[depth][i]]] = LIGHT;
if (ask(w) < expected) high = (pos = mid) - 1;
else low = mid + 1;
}
answer(0, ve[depth][pos]);
}
#ifdef hwe
signed main() {
ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
cerr << '\n'; return 0;
}
#endif
Compilation message
highway.cpp: In function 'void find_pair(int, std::vector<int>, std::vector<int>, int, int)':
highway.cpp:104:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
104 | int mid = low + high >> 1; vector<int> w(m, HEAVY); ll expected = 1ll * mid * A;
| ~~~~^~~~~~
highway.cpp:114:17: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
114 | int mid = low + high >> 1; vector<int> w(m, HEAVY);
| ~~~~^~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
4688 KB |
Output is incorrect: {s, t} is wrong. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
4688 KB |
Output is incorrect: {s, t} is wrong. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
12 ms |
6336 KB |
Output is incorrect: {s, t} is wrong. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
2 ms |
4688 KB |
Output is incorrect: {s, t} is wrong. |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
77 ms |
33340 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Runtime error |
68 ms |
33352 KB |
Execution killed with signal 6 |
2 |
Halted |
0 ms |
0 KB |
- |