답안 #890051

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
890051 2023-12-20T12:35:33 Z Onown 공장들 (JOI14_factories) C++17
33 / 100
8000 ms 231996 KB
#include <bits/stdc++.h>
#include <factories.h>
using namespace std;
 
const int N = 5e5 + 55, L = 2e0 + 20, Q = 30;
const long long I = 1e18 + 118;
int n, q, h[N], par[N][L];
long long spt[N][L], dis[N];
vector<pair<int, int>> adj[N];
 
void dfs(int v, int p) {
	for (auto [u, w]: adj[v])
		if (u != p) {
			par[u][0] = v;
			spt[u][0] = w;
			h[u] = h[v] + 1;
			dfs(u, v);
		}
}
 
void dfsu(int v, int p) {
	for (auto [u, w]: adj[v])
		if (u != p) {
			dfsu(u, v);
			dis[v] = min(dis[v], dis[u] + w);
		}
}
 
void dfsd(int v, int p) {
	for (auto [u, w]: adj[v])
		if (u != p) {
			dis[u] = min(dis[u], dis[v] + w);
			dfsd(u, v);
		}
}
 
long long distance(int v, int u) {
	long long ans = 0;
	if (h[v] < h[u]) swap(v, u);
	for (int i = 0; i < L; i++)
		if ((1 << i) & (h[v] - h[u])) {
			ans += spt[v][i];
			v = par[v][i];
		}
	if (v == u) return ans;
 
	for (int i = L - 1; i >= 0; i--)
		if (par[v][i] != par[u][i]) {
			ans += spt[v][i] + spt[u][i];
			v = par[v][i];
			u = par[u][i];
		}
	return ans + spt[v][0] + spt[u][0];
}
 
void Init(int N, int A[], int B[], int D[]) {
	n = N;
	for (int i = 0; i < n - 1; i++) {
		adj[A[i] + 1].push_back({B[i] + 1, D[i]});
		adj[B[i] + 1].push_back({A[i] + 1, D[i]});
	}
 
	dfs(1, 0);
	for (int i = 1; i < L; i++)
		for (int j = 1; j <= n; j++) {
			par[j][i] = par[par[j][i - 1]][i - 1];
			spt[j][i] = spt[j][i - 1] + spt[par[j][i - 1]][i - 1];
		}
	for (int i = 0; i < N; dis[i++] = I);
}
 
long long Query(int S, int X[], int T, int Y[]) {
	if (S < Q) {
		long long ans = I;
		for (int i = 0; i < T; i++)
			for (int j = 0; j < S; j++)
				ans = min(ans, distance(Y[i] + 1, X[j] + 1));
		return ans;
	}
	
	for (int i = 0; i < S; i++) 
		dis[X[i] + 1] = 0;
	dfsu(1, 0);
	dfsd(1, 0);
 
	long long ans = I;
	for (int i = 0; i < T; i++) ans = min(ans, dis[Y[i] + 1]);
	for (int i = 0; i < N; dis[i++] = I);
	return ans;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 75 ms 37724 KB Output is correct
2 Correct 620 ms 51460 KB Output is correct
3 Correct 2009 ms 51336 KB Output is correct
4 Correct 583 ms 51324 KB Output is correct
5 Correct 637 ms 51536 KB Output is correct
6 Correct 435 ms 51332 KB Output is correct
7 Correct 1963 ms 51344 KB Output is correct
8 Correct 1646 ms 51536 KB Output is correct
9 Correct 646 ms 51796 KB Output is correct
10 Correct 427 ms 51320 KB Output is correct
11 Correct 1931 ms 51580 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 35420 KB Output is correct
2 Correct 1241 ms 207636 KB Output is correct
3 Correct 3507 ms 209748 KB Output is correct
4 Correct 834 ms 208308 KB Output is correct
5 Correct 3567 ms 231996 KB Output is correct
6 Correct 2492 ms 210620 KB Output is correct
7 Correct 4581 ms 85300 KB Output is correct
8 Correct 994 ms 85280 KB Output is correct
9 Correct 4959 ms 88080 KB Output is correct
10 Correct 2767 ms 85912 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 75 ms 37724 KB Output is correct
2 Correct 620 ms 51460 KB Output is correct
3 Correct 2009 ms 51336 KB Output is correct
4 Correct 583 ms 51324 KB Output is correct
5 Correct 637 ms 51536 KB Output is correct
6 Correct 435 ms 51332 KB Output is correct
7 Correct 1963 ms 51344 KB Output is correct
8 Correct 1646 ms 51536 KB Output is correct
9 Correct 646 ms 51796 KB Output is correct
10 Correct 427 ms 51320 KB Output is correct
11 Correct 1931 ms 51580 KB Output is correct
12 Correct 8 ms 35420 KB Output is correct
13 Correct 1241 ms 207636 KB Output is correct
14 Correct 3507 ms 209748 KB Output is correct
15 Correct 834 ms 208308 KB Output is correct
16 Correct 3567 ms 231996 KB Output is correct
17 Correct 2492 ms 210620 KB Output is correct
18 Correct 4581 ms 85300 KB Output is correct
19 Correct 994 ms 85280 KB Output is correct
20 Correct 4959 ms 88080 KB Output is correct
21 Correct 2767 ms 85912 KB Output is correct
22 Correct 6033 ms 212456 KB Output is correct
23 Correct 3966 ms 213720 KB Output is correct
24 Correct 7513 ms 215244 KB Output is correct
25 Correct 7164 ms 217636 KB Output is correct
26 Execution timed out 8061 ms 215912 KB Time limit exceeded
27 Halted 0 ms 0 KB -