Submission #1359615

#TimeUsernameProblemLanguageResultExecution timeMemory
1359615AMel0nHighway Tolls (IOI18_highway)C++20
5 / 100
236 ms327680 KiB
#include "highway.h"
#include <bits/stdc++.h>
using namespace std; 

vector<int> E, P;
vector<vector<pair<int,int>>> adj;
void dfs1(int v, int p) {
	for(auto &[u, i]: adj[v]) {
		if (u == p) continue;
		E[i] = 1;
		dfs1(u, v);
	}
}
void dfs2(int v, int p, int d, int t) {
	if (d == t) P.push_back(v);
	for(auto &[u, i]: adj[v]) {
		if (u == p) continue;
		dfs2(u, v, d+1, t);
	}
}

void find_pair(int N, vector<int> U, vector<int> V, int A, int B) {
  	int M = U.size();
	E.resize(M), adj.resize(N);
	for(int i = 0; i < M; i++) adj[U[i]].push_back({V[i], i}), adj[V[i]].push_back({U[i], i});
	int S, T;

	int emptc = ask(E);
	int l = 0, r = M;
	while(l < r - 1) {
		int m = (l + r) / 2;
		for(int i = 0; i < m; i++) E[i] = 1;
		for(int i = m; i < M; i++) E[i] = 0;
		if (ask(E) == emptc) l = m;
		else r = m;
	}
	int u = U[l], v = V[l];

	E.assign(M, 0);
	dfs1(u, v); 
	int nume = (ask(E) - emptc) / (B-A);
	dfs2(u, v, 0, nume);
	l = 0, r = P.size();
	while(l < r - 1) {
		int m = (l + r) / 2;
		E.assign(M, 0);
		for(int i = 0; i < m; i++) {
			for(auto &[a, i]: adj[P[i]]) E[i] = 1;
		}
		if (ask(E) != emptc) r = m;
		else l = m;
	}
	S = P[l];

	P.clear();
	E.assign(M, 0);
	dfs1(v, u);
	nume = (ask(E) - emptc) / (B-A);
	dfs2(v, u, 0, nume);
	l = 0, r = P.size();
	while(l < r - 1) {
		int m = (l + r) / 2;
		E.assign(M, 0);
		for(int i = 0; i < m; i++) {
			for(auto &[a, i]: adj[P[i]]) E[i] = 1;
		}
		if (ask(E) != emptc) r = m;
		else l = m;
	}
	T = P[l];
	answer(S, T);
}
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...