結果

問題 No.3113 The farthest point
コンテスト
ユーザー magurofly
提出日時 2025-04-19 01:51:00
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,647 bytes
コンパイル時間 15,513 ms
コンパイル使用メモリ 400,400 KB
実行使用メモリ 35,736 KB
最終ジャッジ日時 2025-04-19 01:51:33
合計ジャッジ時間 19,159 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

use proconio::{input, marker::Usize1};

fn main() {
    input! {
        n: usize,
        edges: [(Usize1, Usize1, i64); n - 1],
    }

    let mut graph = vec![vec![]; n];
    for &(u, v, w) in &edges {
        graph[u].push((v, w));
        graph[v].push((u, w));
    }

    let dist0 = dfs_dist(&graph, 0);

    fn dfs(graph: &[Vec<(usize, i64)>], dist: &[i64], u: usize, p: usize, depth: usize, down: &mut [i64], up: &mut [i64], ans: &mut i64) {
        *ans = (*ans).max(down[depth]);
        up[depth] = 0;
        for &(v, w) in &graph[u] {
            if v == p {
                continue;
            }
            down[depth + 1] = down[depth] + w;
            up[depth + 1] = -INF;
            dfs(graph, dist, v, u, depth + 1, down, up, ans);
            up[depth] = up[depth].max(w + up[depth + 1]);
            down[depth] = down[depth].max(up[depth]);
        }
    }

    let mut ans = 0;
    for _ in 0 .. 2 {
        let mut down = vec![-INF; n];
        down[0] = 0;
        let mut up = vec![-INF; n];
        dfs(&graph, &dist0, 0, 0, 0, &mut down, &mut up, &mut ans);
        for v in 0 .. n {
            graph[v].reverse();
        }
    }
    println!("{ans}");
}

const INF: i64 = 1_000_000_000_000_000_000;

fn dfs_dist(graph: &[Vec<(usize, i64)>], start: usize) -> Vec<i64> {
    let mut dist = vec![INF; graph.len()];
    dist[start] = 0;
    let mut stack = vec![(start, start)];
    while let Some((u, p)) = stack.pop() {
        for &(v, w) in &graph[u] {
            if v == p {
                continue;
            }
            dist[v] = dist[u] + w;
            stack.push((v, u));
        }
    }
    dist
}
0