結果
問題 | No.994 ばらばらコイン |
ユーザー | phspls |
提出日時 | 2020-02-21 22:25:16 |
言語 | Rust (1.72.1) |
結果 |
AC
|
実行時間 | 59 ms / 2,000 ms |
コード長 | 1,644 bytes |
コンパイル時間 | 1,287 ms |
コンパイル使用メモリ | 160,712 KB |
実行使用メモリ | 11,888 KB |
最終ジャッジ日時 | 2023-07-30 05:56:12 |
合計ジャッジ時間 | 3,078 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,376 KB |
testcase_02 | AC | 27 ms
8,820 KB |
testcase_03 | AC | 40 ms
10,392 KB |
testcase_04 | AC | 40 ms
9,864 KB |
testcase_05 | AC | 20 ms
6,324 KB |
testcase_06 | AC | 59 ms
11,888 KB |
testcase_07 | AC | 39 ms
9,868 KB |
testcase_08 | AC | 24 ms
6,960 KB |
testcase_09 | AC | 34 ms
9,604 KB |
testcase_10 | AC | 25 ms
7,432 KB |
testcase_11 | AC | 42 ms
10,092 KB |
testcase_12 | AC | 36 ms
9,388 KB |
testcase_13 | AC | 1 ms
4,380 KB |
testcase_14 | AC | 1 ms
4,384 KB |
testcase_15 | AC | 1 ms
4,376 KB |
testcase_16 | AC | 1 ms
4,376 KB |
testcase_17 | AC | 1 ms
4,376 KB |
testcase_18 | AC | 1 ms
4,376 KB |
testcase_19 | AC | 1 ms
4,376 KB |
testcase_20 | AC | 1 ms
4,380 KB |
testcase_21 | AC | 1 ms
4,380 KB |
testcase_22 | AC | 1 ms
4,380 KB |
ソースコード
use std::cmp::min; use std::collections::HashSet; fn main() { let mut nk = String::new(); std::io::stdin().read_line(&mut nk).ok(); let nk: Vec<usize> = nk.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let n = nk[0]; let k = nk[1]; let mut ab: Vec<Vec<usize>> = vec![vec![]; n]; for _ in 0..n-1 { let mut aab = String::new(); std::io::stdin().read_line(&mut aab).ok(); let aab: Vec<usize> = aab.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); ab[aab[0] - 1].push(aab[1] - 1); ab[aab[1] - 1].push(aab[0] - 1); } if n < k { println!("{}", -1); return; } if 0 == k { println!("{}", 0); return; } let mut dists: Vec<usize> = vec![1000000; n]; dists[0] = 0; let mut checked: HashSet<usize> = HashSet::new(); checked.insert(0); let mut vertices: Vec<usize> = vec![]; vertices.push(0); let mut dist: usize = 1; let mut result: usize = 0; while vertices.len() > 0 { let mut nexts: Vec<usize> = vec![]; vertices.iter().for_each(|v| { if checked.len() == k { return; } ab[*v].iter().for_each(|i| { if checked.len() == k || checked.contains(i) { return; } nexts.push(*i); checked.insert(*i); dists[*i] = min(dists[*i], dist); result += 1; }); }); if checked.len() == k { break; } vertices = nexts; dist += 1; } println!("{}", result); }