結果
問題 | No.1103 Directed Length Sum |
ユーザー | fukafukatani |
提出日時 | 2021-01-12 20:56:57 |
言語 | Rust (1.77.0 + proconio) |
結果 |
AC
|
実行時間 | 1,011 ms / 3,000 ms |
コード長 | 1,430 bytes |
コンパイル時間 | 13,382 ms |
コンパイル使用メモリ | 378,048 KB |
実行使用メモリ | 205,112 KB |
最終ジャッジ日時 | 2024-05-01 07:34:29 |
合計ジャッジ時間 | 23,667 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
6,812 KB |
testcase_01 | AC | 1 ms
6,816 KB |
testcase_02 | AC | 493 ms
205,112 KB |
testcase_03 | AC | 309 ms
111,316 KB |
testcase_04 | AC | 545 ms
77,336 KB |
testcase_05 | AC | 1,011 ms
132,396 KB |
testcase_06 | AC | 332 ms
50,944 KB |
testcase_07 | AC | 58 ms
13,824 KB |
testcase_08 | AC | 104 ms
19,968 KB |
testcase_09 | AC | 32 ms
9,344 KB |
testcase_10 | AC | 154 ms
26,624 KB |
testcase_11 | AC | 600 ms
84,076 KB |
testcase_12 | AC | 344 ms
51,456 KB |
testcase_13 | AC | 158 ms
27,392 KB |
testcase_14 | AC | 24 ms
7,424 KB |
testcase_15 | AC | 293 ms
40,456 KB |
testcase_16 | AC | 689 ms
94,336 KB |
testcase_17 | AC | 722 ms
98,304 KB |
testcase_18 | AC | 151 ms
26,560 KB |
testcase_19 | AC | 613 ms
86,272 KB |
testcase_20 | AC | 39 ms
10,880 KB |
testcase_21 | AC | 88 ms
18,304 KB |
testcase_22 | AC | 493 ms
70,700 KB |
testcase_23 | AC | 279 ms
43,244 KB |
コンパイルメッセージ
warning: unused variable: `i` --> src/main.rs:9:9 | 9 | for i in 0..n - 1 { | ^ help: if this is intentional, prefix it with an underscore: `_i` | = note: `#[warn(unused_variables)]` on by default
ソースコード
#![allow(unused_imports)] use std::cmp::*; use std::collections::*; fn main() { let n = read::<usize>(); let mut edges = vec![vec![]; n]; let mut rev_edges = vec![vec![]; n]; for i in 0..n - 1 { let v = read_vec::<usize>(); let (a, b) = (v[0] - 1, v[1] - 1); edges[a].push(b); rev_edges[b].push(a); } let root = (0..n).find(|&x| rev_edges[x].is_empty()).unwrap(); let mut pre_order = vec![]; dfs(root, &edges, &mut pre_order); pre_order.reverse(); let mut count = vec![0; n]; let mut dp = vec![0; n]; for ti in pre_order { if edges[ti].is_empty() { continue; } for &ci in &edges[ti] { count[ti] += 1 + count[ci]; dp[ti] += dp[ci]; } dp[ti] += count[ti]; dp[ti] %= 1000000007; } let ans = dp.iter().fold(0, |s, x| (s + x) % 1000000007); println!("{}", ans); } fn dfs(cur: usize, edges: &Vec<Vec<usize>>, pre_order: &mut Vec<usize>) { pre_order.push(cur); for &to in edges[cur].iter() { dfs(to, edges, pre_order); } } fn read<T: std::str::FromStr>() -> T { let mut s = String::new(); std::io::stdin().read_line(&mut s).ok(); s.trim().parse().ok().unwrap() } fn read_vec<T: std::str::FromStr>() -> Vec<T> { read::<String>() .split_whitespace() .map(|e| e.parse().ok().unwrap()) .collect() }