oj.uz

답안 #996835

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
996835 2024-06-11T10:19:31 Z alexdumitru 경주 (Race) (IOI11_race) C++14
0 / 100
 3000 ms 8540 KB 
#include "race.h"
#include <iostream>
#include <vector>
 
using namespace std;
 
const int NMAX = 1e5;
const int KMAX = 1e6;
 
vector<pair<int, int>> g[NMAX];
int w[NMAX];
bool used[NMAX];
int minLength[KMAX + 1];
int Ans;
 
int get_w(int node, int dad = -1) {
    w[node] = 1;
    for (auto it : g[node]) {
        if (it.first == dad || used[it.first]) continue;
        get_w(it.first, node);
        w[node] += w[it.first];
    }
    return w[node];
}
 
int get_centroid(int node, int totalW, int dad = -1) {
    for (auto it : g[node]) {
        if (it.first == dad || used[it.first]) continue;
        if (w[it.first] > totalW / 2)
            return get_centroid(it.first, node);
    }
    return node;
}
 
void dfs_add(int node, int k, int length = 0, int dad = -1, int depth = 0) {
    if (length > k) return;
    minLength[length] = min(minLength[length], depth);
 
    for (auto it : g[node]) {
        if (it.first == dad || used[it.first]) continue;
        dfs_add(it.first, k, length + it.second, node, depth + 1);
    }
}
 
void dfs_compute(int node, int k, int length = 0, int dad = -1) {
    if (length > k) return;
    Ans = min(Ans, minLength[length] + minLength[k - length]);
 
    for (auto it : g[node]) {
        if (it.first == dad || used[it.first]) continue;
        dfs_compute(it.first, k, length + it.second, node);
    }
}
 
void dfs_clear(int node, int k, int length = 0, int dad = -1) {
    if (length > k) return;
    minLength[length] = NMAX;
 
    for (auto it : g[node]) {
        if (it.first == dad || used[it.first]) continue;
        dfs_clear(it.first, k, length + it.second, node);
    }
}
 
void go(int node, int k) {
    int totalW = get_w(node);
    int c = get_centroid(node, totalW);
    used[c] = true;
 
  	dfs_clear(c, k);
    dfs_add(c, k);
    dfs_compute(c, k);
    dfs_clear(c, k);
 
    for (auto it : g[c])
        if (!used[it.first])
            go(it.first, k);
}
 
int best_path(int N, int K, int H[][2], int L[]) {
    Ans = N;
  
    for (int i = 0; i < N - 1; i++) {
        g[H[i][0]].emplace_back(H[i][1], L[i]);
        g[H[i][1]].emplace_back(H[i][0], L[i]);
    }
 
    go(0, K);
 
    return Ans == N ? -1 : Ans;
}

Subtask #1
0 / 9.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 3037 ms 8540 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #2
0 / 12.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 3037 ms 8540 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #3
0 / 22.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 3037 ms 8540 KB Time limit exceeded
2 Halted 0 ms 0 KB -

Subtask #4
0 / 57.0

# 결과 실행 시간 메모리 Grader output
1 Execution timed out 3037 ms 8540 KB Time limit exceeded
2 Halted 0 ms 0 KB -