結果
問題 | No.994 ばらばらコイン |
ユーザー |
|
提出日時 | 2020-02-21 22:25:16 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 54 ms / 2,000 ms |
コード長 | 1,644 bytes |
コンパイル時間 | 13,385 ms |
コンパイル使用メモリ | 403,096 KB |
実行使用メモリ | 11,956 KB |
最終ジャッジ日時 | 2024-10-08 21:56:59 |
合計ジャッジ時間 | 15,097 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 23 |
ソースコード
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); }