oj.uz

답안 #1090123

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
1090123 2024-09-17T18:26:44 Z LilPluton 경주 (Race) (IOI11_race) C++14
0 / 100
 7 ms 14424 KB 
#include "race.h"
#include <bits/stdc++.h>
using namespace std;

const int sz = 2e5 + 5;
int n, res, k;
vector<vector<pair<int, int>>> adj(sz);
map<int, int> a[sz];
long long sum[sz];
int dep[sz];

void comp(int u, int p, long long  c, int h) {
	a[u][c] = h;
	sum[u] = c;
	dep[u] = h;
	for (auto n : adj[u]) {
		if (p == n.first) { continue; }
		comp(n.first, u, c + n.second, h + 1);
	}
}

void dfs(int v, int p) {
    int t = k + 2 * sum[v];
    for (auto i : adj[v]) {
        if (i.first == p) {
            continue;
        }
        if (a[i.first].size() > a[v].size()) {
            swap(a[i.first], a[v]);
        }
        for (auto j : a[i.first]) {
            if (a[v].find(t - j.first) != a[v].end()) {
                res = min(res, a[v][t - j.first] + j.second - 2 * dep[v]);
            }
        }
        for (auto j : a[i.first]) {
            if (a[v].find(j.first) == a[v].end()) {
                a[v].insert(j);
            } else {
                a[v][j.first] = min(a[v][j.first], j.second);
            }
        }
    }
}

int best_path(int N, int K, int H[][2], int L[]) {
    if (K == 1) {
        return 0;
    }
    n = N;
    k = K;
    for (int i = 0; i < N; ++i) {
        adj[i].clear();
        a[i].clear();
    }
    
    for (int i = 0; i < N - 1; ++i) {
        adj[H[i][0]].push_back({H[i][1], L[i]});
        adj[H[i][1]].push_back({H[i][0], L[i]});
    }
    
    res = INT_MAX;
    comp(0, -1, 0, 0);
    dfs(0, -1);
    
    return (res == INT_MAX ? -1 : res);
}

Subtask #1
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14424 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14424 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14424 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #4
0 / 57.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 7 ms 14424 KB Output isn't correct
2 Halted 0 ms 0 KB -