結果
問題 | No.1098 LCAs |
ユーザー |
|
提出日時 | 2022-12-29 02:19:28 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 468 ms / 2,000 ms |
コード長 | 1,056 bytes |
コンパイル時間 | 16,347 ms |
コンパイル使用メモリ | 378,560 KB |
実行使用メモリ | 40,704 KB |
最終ジャッジ日時 | 2024-11-24 04:45:10 |
合計ジャッジ時間 | 22,756 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
fn dfs(cur: usize, prev: usize, paths: &Vec<Vec<usize>>, result: &mut Vec<usize>) -> usize {let mut children = 1usize;let mut cs = vec![1];for &v in paths[cur].iter() {if v == prev { continue; }let val = dfs(v, cur, paths, result);children += val;cs.push(val);}result[cur] = children * children - cs.iter().map(|&v| v * v).sum::<usize>() + 1;children}fn main() {let mut n = String::new();std::io::stdin().read_line(&mut n).ok();let n: usize = n.trim().parse().unwrap();let mut paths = vec![vec![]; n];for _ in 0..n-1 {let mut temp = String::new();std::io::stdin().read_line(&mut temp).ok();let temp: Vec<usize> = temp.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();let u = temp[0]-1;let v = temp[1]-1;paths[u].push(v);paths[v].push(u);}let mut result = vec![0usize; n];dfs(0, 0, &paths, &mut result);for &v in result.iter() {println!("{}", v);}}