제출 #170845

#제출 시각아이디문제언어결과실행 시간메모리
170845ZwariowanyMarcin공장들 (JOI14_factories)C++14
33 / 100
8086 ms223336 KiB
#include "factories.h"
#include <bits/stdc++.h>
#define fi first
#define se second
#define mp make_pair
#define ss(x) (int) x.size()
#define pb push_back
#define ll long long
#define cat(x) cerr << #x << " = " << x << endl
#define FOR(i, n) for(int i = 0; i < n; ++i)

using namespace std;

const int nax = 5e5 + 111;

int n;
vector <pair<int,int>> v[nax];
int par[nax];
int kil[nax];
int nn;
int siz[nax];
int h[nax];
int jump[nax][20];
ll H[nax];

void prep(int u, int p) {
	jump[u][0] = p;
	h[u] = h[p] + 1;
	for(auto it : v[u])
		if(it.fi != p) {
			H[it.fi] = H[u] + it.se;
			prep(it.fi, u);
		}
}

int lca(int x, int y) {
	if(h[x] < h[y])
		swap(x, y);
	for(int i = 19; 0 <= i; --i)
		if(h[y] <= h[x] - (1 << i))
			x = jump[x][i];
	if(x == y)
		return x;
	for(int i = 19; 0 <= i; --i)
		if(jump[x][i] != jump[y][i]) {
			x = jump[x][i];
			y = jump[y][i];
		}
	return jump[x][0];
}
		
ll dis(int x, int y) {
	return H[x] + H[y] - 2 * H[lca(x, y)];
}

void dfs(int u, int p) {
	nn++;
	siz[u] = 1;
	for(auto it : v[u])
		if(it.fi != p && !kil[it.fi]) {
			dfs(it.fi, u);
			siz[u] += siz[it.fi];
		}
}

int daj(int u, int p) {
	for(auto it : v[u])
		if(it.fi != p && !kil[it.fi] && nn <= 2 * siz[it.fi])
			return daj(it.fi, u);
	return u;
}

void decomp(int u, int p) {
	nn = 0;
	dfs(u, 0);
	int C = daj(u, 0);
	par[C] = p;
	kil[C] = 1;
	for(auto it : v[C])
		if(!kil[it.fi])
			decomp(it.fi, C);
}

ll naj[nax];
ll odl[nax][20];

void Init(int N, int a[], int b[], int c[]) {
	n = N;
	for(int i = 0; i < n; ++i)
		a[i]++;
	for(int i = 0; i < n; ++i)
		b[i]++;
	for(int i = 0; i < n - 1; ++i) {
		v[a[i]].pb(mp(b[i], c[i]));
		v[b[i]].pb(mp(a[i], c[i]));
	}
	prep(1, 0);
	decomp(1, 0);
	for(int j = 1; j <= 19; ++j)
		for(int i = 1; i <= n; ++i)
			jump[i][j] = jump[jump[i][j - 1]][j - 1];
	for(int i = 1; i <= n; ++i)
		naj[i] = 1e18;
	for(int i = 1; i <= n; ++i) {
		int lvl = 0;
		int node = i;
		while(node != 0) {
			odl[i][lvl] = dis(i, node);
			lvl++;
			node = par[node];
		}
	}
}

ll Query(int s, int x[], int t, int y[]) {
	for(int i = 0; i < s; ++i)
		x[i]++;
	for(int i = 0; i < t; ++i)
		y[i]++;
	ll res = 1e18;
	for(int i = 0; i < s; ++i) {
		int node = x[i];
		int lvl = 0;
		while(node != 0) {
			naj[node] = min(naj[node], odl[x[i]][lvl]);
			node = par[node];
			lvl++;
		}
	}
	for(int i = 0; i < t; ++i) {
		int node = y[i];
		int lvl = 0;
		while(node != 0) {
			res = min(res, naj[node] + odl[y[i]][lvl]);
			node = par[node];
			lvl++;
		}
	}
	for(int i = 0; i < s; ++i) {
		int node = x[i];
		while(node != 0) {
			naj[node] = 1e18;
			node = par[node];
		}
	}
	return res;
}	
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...