Submission #97164

#TimeUsernameProblemLanguageResultExecution timeMemory
97164youngyojunHighway Tolls (IOI18_highway)C++11
100 / 100
487 ms26320 KiB
#include "highway.h"
#include <bits/stdc++.h>
#define eb emplace_back
#define sz(V) ((int)(V).size())
#define allv(V) ((V).begin()),((V).end())
using namespace std;
typedef long long ll;

const int MAXN = 90055;
const int MAXM = 130055;

vector<int> G[MAXN], BEV;

vector<int> PEV[2][MAXN], O[2];
int C[2][MAXN], D[2][MAXN];
int A[MAXM], B[MAXM];

int ud[MAXN];

int N, M, CA, CB, L, W, AS, AT;

int uf(int i) { return i == ud[i] ? i : (ud[i] = uf(ud[i])); }
void uf(int a, int b) { ud[uf(b)] = uf(a); }

void getW() {
	int s = 0, e = M-1; for(int m; s < e;) {
		m = (s+e) >> 1;
		vector<int> V(M, 0);
		fill(V.begin(), V.begin()+m+1, 1);
		if(ll(L)*CA < ask(V)) e = m;
		else s = m+1;
	}
	W = s;
}

void calC() {
	for(int i = 0; i < M; i++) {
		G[A[i]].eb(i);
		G[B[i]].eb(i);
	}
	for(int t : {0, 1}) {
		queue<int> PQ;
		C[t][(t ? B : A)[W]] = 1;
		PQ.push((t ? B : A)[W]);
		for(int i, d; !PQ.empty(); PQ.pop()) {
			i = PQ.front();
			d = C[t][i] + 1;
			for(int e : G[i]) {
				int v = A[e] + B[e] - i;
				if(C[t][v]) continue;
				C[t][v] = d;
				PQ.push(v);
			}
		}
	}
}

void dfs(int D[], vector<int> PEV[], int i) {
	for(int e : G[i]) {
		int v = A[e] + B[e] - i;
		PEV[v].eb(e);
		if(D[v]) continue;
		D[v] = D[i] + 1;
		dfs(D, PEV, v);
	}
}

void findS() {
	int s = 0, e = sz(O[0])-1; for(int m; s < e;) {
		m = (s+e) >> 1;
		vector<int> V = BEV;
		for(int i = 0; i <= m; i++)
			V[PEV[0][O[0][i]][0]] = 1;
		if(ll(L)*CA < ask(V)) e = m;
		else s = m+1;
	}
	AS = O[0][s];
}

void findT() {
	vector<int> HV;
	int dep = L-D[0][AS]+1;
	for(int i = 0; i < N; i++) if(dep == D[1][i]) HV.eb(i);
	int s = 0, e = sz(HV)-1; for(int m; s < e;) {
		m = (s+e) >> 1;
		vector<int> V = BEV;
		for(int i = 0; i <= m; i++)
			V[PEV[1][HV[i]][0]] = 1;
		if(ll(L)*CA < ask(V)) e = m;
		else s = m+1;
	}
	AT = HV[s];
}

void solve() {
	L = ask(vector<int>(M, 0)) / CA;
	getW();
	calC();
	for(int i = 0; i < N; i++) G[i].clear();
	for(int i = 0, a, b; i < M; i++) {
		a = A[i]; b = B[i];
		if(C[0][a] == C[0][b] || C[1][a] == C[1][b]
			|| C[0][a] == C[1][a] || C[0][b] == C[1][b]
			|| (C[0][a] < C[0][b]) != (C[1][a] < C[1][b])) {
			BEV.eb(1);
			continue;
		}
		BEV.eb(0);
		if(C[0][a] < C[0][b]) G[a].eb(i);
		else G[b].eb(i);
	}
	BEV[W] = 0;
	for(int t : {0, 1}) {
		int v = (t ? B : A)[W];
		D[t][v] = 1; PEV[t][v].eb(W);
		dfs(D[t], PEV[t], v);
		for(int i = 0; i < N; i++)
			if(D[t][i]) O[t].eb(i);
		sort(allv(O[t]), [&](int a, int b) {
			return D[t][a] > D[t][b];
		});
		iota(ud, ud+N, 0);
		for(int v : O[t]) {
			auto &V = PEV[t][v];
			int pe = -1;
			for(int e : V) {
				int p = A[e] + B[e] - v;
				if(uf(p) == uf(v)) continue;
				uf(p, v);
				pe = e;
				break;
			}
			for(int e : V) if(e != pe) BEV[e] = 1;
			V.clear(); V.eb(pe);
		}
	}
	findS();
	findT();
	answer(AS, AT);
}

void find_pair(int N, std::vector<int> U, std::vector<int> V, int A, int B) {
	::N = N; ::M = U.size(); ::CA = A; ::CB = B;
	for(int i = 0; i < ::M; i++) {
		::A[i] = U[i];
		::B[i] = V[i];
	}
	solve();
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...