結果
問題 |
No.1098 LCAs
|
ユーザー |
|
提出日時 | 2023-12-14 23:21:06 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 338 ms / 2,000 ms |
コード長 | 2,311 bytes |
コンパイル時間 | 11,265 ms |
コンパイル使用メモリ | 388,624 KB |
実行使用メモリ | 34,496 KB |
最終ジャッジ日時 | 2024-09-27 05:50:06 |
合計ジャッジ時間 | 16,824 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
#![allow(non_snake_case)] #![allow(unused_imports)] //use itertools::itertools; //use proconio::input; //use proconio::marker::*; use std::cmp::Reverse; use std::collections::*; fn main() { input! { n:usize, uv:[(usize1,usize1);n-1], } let mut g = vec![vec![]; n]; for (u, v) in uv { g[u].push(v); g[v].push(u); } let mut ans = vec![0; n]; f(0, n, &g, &mut ans); for i in 0..n { println!("{}", ans[i]); } } fn f(crr: usize, pre: usize, g: &Vec<Vec<usize>>, ans: &mut Vec<usize>) -> (usize, usize) { let mut childs = 0; let mut cnts = 0; for &nex in g[crr].iter() { if nex == pre { continue; } let (child, cnt) = f(nex, crr, g, ans); childs += child; cnts += cnt; } ans[crr] = (childs + 1) * (childs + 1) - cnts; (childs + 1, ans[crr] + cnts) } mod input { #[macro_export] macro_rules! input { (source = $s:expr, $($r:tt)*) => { let mut iter = $s.split_whitespace(); input_inner!{iter, $($r)*} }; ($($r:tt)*) => { let s = { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); s }; let mut iter = s.split_whitespace(); input_inner!{iter, $($r)*} }; } #[macro_export] macro_rules! input_inner { ($iter:expr) => {}; ($iter:expr, ) => {}; ($iter:expr, $var:ident : $t:tt $($r:tt)*) => { let $var = read_value!($iter, $t); input_inner!{$iter $($r)*} }; } #[macro_export] macro_rules! read_value { ($iter:expr, ( $($t:tt),* )) => { ( $(read_value!($iter, $t)),* ) }; ($iter:expr, [ $t:tt ; $len:expr ]) => { (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>() }; ($iter:expr, chars) => { read_value!($iter, String).chars().collect::<Vec<char>>() }; ($iter:expr, usize1) => { read_value!($iter, usize) - 1 }; ($iter:expr, $t:ty) => { $iter.next().unwrap().parse::<$t>().expect("Parse error") }; } }