| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 255264 | Erkhemkhuu | 통행료 (IOI18_highway) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#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 = 100005;
pair <int, int> par[N];
vector <bool> w;
vector <vector <pair <int, int> > > adj(N);
int flat[N], timer;
void dfs(ll v, ll 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 find_pair(int n, vector <int> u, vector <int> v, int a, int b) {
int i, m, max_cost, l, r, mid, s, t;
m = u.size();
for(i = 0; i < m; i++) {
adj[u[i]].pb({v[i], i});
adj[v[i]].pb({u[i], i});
w[i] = true;
}
dfs(0, -1);
max_cost = ask(w);
w.resize(m, false);
l = 0; r = m;
while(l + 1 < r) {
mid = (l + r) / 2;
for(i = l; i <= mid; i++)
w[par[flat[i]].S] = true;
if(ask(w) >= max_cost) {
r = mid;
for(i = l; i <= mid; i++)
w[par[flat[i]].S] = false;
}
else l = mid;
}
s = flat[l];
w.resize(m, false);
dfs(s, -1);
l = 1; r = m;
while(l + 1 < r) {
mid = (l + r) / 2;
for(i = l; i <= mid; i++)
w[par[flat[i]].S] = true;
if(ask(w) >= max_cost) {
r = mid;
for(i = l; i <= mid; i++)
w[par[flat[i]].S] = false;
}
else l = mid;
}
t = flat[l];
answer(s, t);
return;
}