結果
問題 | No.1103 Directed Length Sum |
ユーザー | fukafukatani |
提出日時 | 2021-01-12 20:56:57 |
言語 | Rust (1.77.0 + proconio) |
結果 |
AC
|
実行時間 | 1,133 ms / 3,000 ms |
コード長 | 1,430 bytes |
コンパイル時間 | 12,652 ms |
コンパイル使用メモリ | 398,324 KB |
実行使用メモリ | 205,056 KB |
最終ジャッジ日時 | 2024-11-21 10:17:25 |
合計ジャッジ時間 | 23,072 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,248 KB |
testcase_02 | AC | 492 ms
205,056 KB |
testcase_03 | AC | 307 ms
111,360 KB |
testcase_04 | AC | 573 ms
77,440 KB |
testcase_05 | AC | 1,133 ms
132,352 KB |
testcase_06 | AC | 368 ms
50,944 KB |
testcase_07 | AC | 68 ms
13,824 KB |
testcase_08 | AC | 114 ms
19,840 KB |
testcase_09 | AC | 38 ms
9,216 KB |
testcase_10 | AC | 165 ms
26,496 KB |
testcase_11 | AC | 689 ms
84,096 KB |
testcase_12 | AC | 363 ms
51,328 KB |
testcase_13 | AC | 171 ms
27,392 KB |
testcase_14 | AC | 28 ms
7,552 KB |
testcase_15 | AC | 279 ms
40,576 KB |
testcase_16 | AC | 734 ms
94,336 KB |
testcase_17 | AC | 770 ms
98,304 KB |
testcase_18 | AC | 157 ms
26,440 KB |
testcase_19 | AC | 652 ms
86,144 KB |
testcase_20 | AC | 47 ms
10,752 KB |
testcase_21 | AC | 96 ms
18,396 KB |
testcase_22 | AC | 522 ms
70,784 KB |
testcase_23 | AC | 293 ms
43,252 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() }