# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
594358 | SlavicG | Highway Tolls (IOI18_highway) | C++17 | 0 ms | 0 KiB |
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 "highway.h"
#include "bits/stdc++.h"
using namespace std;
using ll = long long;
#define sz(a) (int)a.size()
void dfs(int u, int par, vector<vector<int>>& adj, vector<int>& p, vector<int>& depth) {
p[u] = par;
for(int v: adj[u]) {
if(v == par) continue;
depth[v] = depth[u] + 1;
dfs(v, u, adj, p, depth);
}
}
int find(int start, int n, vector<int> u, vector<int> v) {
map<pair<int, int>, int> idx;
vector<vector<int>> adj(n);
vector<int> p(n, 0), depth(n, 0);
int m = u.size();
for(int i = 0; i < m; ++i) {
adj[u[i]].push_back(v[i]);
adj[v[i]].push_back(u[i]);
idx[{u[i], v[i]}] = idx[{v[i], u[i]}] = i;
}
dfs(start, -1, adj, p, depth);
int l = 1, r = n, dd = -1;
vector<int> c(m, 1);
ll init = ask(c);
while(l <= r) {
int mid = l + r >> 1;
for(int j = 0; j < n; ++j) {
if(j == start) continue;
if(depth[j] >= mid) {
c[idx[{j, p[j]}]] = 0;
}
}
if(ask(c) < init) {
dd = mid;
l = mid + 1;
} else r = mid - 1;
c.assign(m, 1);
}
vector<int> nodes;
for(int i = 0; i < n; ++i) {
if(i == start) continue;
if(depth[i] == dd) nodes.push_back(i);
}
assert(sz(nodes) > 0);
l = 0, r = sz(nodes) - 1;
int pp = -1;
while(l <= r) {
int mid = l + r >> 1;
for(int i = 0; i <= mid; ++i) {
assert(p[nodes[i]] != -1);
c[idx[{nodes[i], p[nodes[i]]}]] = 0;
}
if(ask(c) < init) {
pp = mid;
r = mid - 1;
} else l = mid + 1;
c.assign(m, 1);
}
assert(pp != -1);
return nodes[pp];
}
void find_pair(int n, vector<int> u, vector<int> v, int a, int b) {
int s = find(0, n, u, v);
int t = find(n - 1 n, u, v);
answer(0, s);
}