Function tree_diameter 

pub fn tree_diameter(
    n: usize,
    edges: &[(usize, usize, u64)],
) -> (u64, Vec<usize>)

木の直径と直径をなす頂点のベクタを返します。