This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
#include <highway.h>
using namespace std;
#define ll long long
#define pb push_back
#define mp make_pair
#define F first
#define S second
const int N = 150005;
pair <int, int> par[N];
vector <int> w;
vector < vector <pair <int, int> > > adj(N);
int flat[N], timer;
void dfs(int v, int p) {
flat[timer++] = v;
for(auto &u: adj[v]) {
if(u.F == p) continue;
par[u.F] = {v, u.S};
dfs(u.F, v);
}
return;
}
void run(int l, int r, int type) {
for(int i = l; i <= r; i++)
w[par[flat[i]].S] = type;
return;
}
int go(int l, int r, int max_cost) {
while(l + 1 < r) {
int mid = (l + r) / 2;
run(l, mid, 1);
if(ask(w) < max_cost) l = mid;
else {
r = mid;
run(l, mid, 0);
}
}
return l + 1;
}
void find_pair(int n, vector <int> u, vector <int> v, int a, int b) {
int i, m, s, t;
ll max_cost;
m = u.size();
w.resize(m);
for(i = 0; i < m; i++) {
adj[u[i]].pb({v[i], i});
adj[v[i]].pb({u[i], i});
}
dfs(0, -1);
w.assign(m, 1);
max_cost = ask(w);
w.assign(m, 0);
s = flat[go(1, n - 1, max_cost)];
w.assign(m, 0);
timer = 0;
memset(par, 0, sizeof(par));
dfs(s, -1);
t = flat[go(1, n - 1, max_cost)];
assert(s >= 0 && s < n && t >= 0 && t < n);
answer(s, t);
return;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |