bouzuya.hatenablog.com

ぼうずやのにっき

鼻水 / ABC214 D

なんだか体調が悪い。また一段と寒くなったからだろうか。鼻水が出る……。

ABC214 : AtCoder Beginner Contest 214

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

fn main() {
    input! {
        n: usize,
        mut uvw: [(Usize1, Usize1, usize); n - 1],
    };
    uvw.sort_by_key(|&(_, _, w)| w);
    let mut sum = 0_usize;
    let mut dsu = Dsu::new(n);
    for (u, v, w) in uvw {
        sum += dsu.size(u) * dsu.size(v) * w;
        dsu.merge(u, v);
    }
    let ans = sum;
    println!("{}", ans);
}

今日のコミット。