結果
問題 |
No.1424 Ultrapalindrome
|
ユーザー |
|
提出日時 | 2022-11-08 23:42:22 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 47 ms / 2,000 ms |
コード長 | 1,555 bytes |
コンパイル時間 | 14,383 ms |
コンパイル使用メモリ | 383,072 KB |
実行使用メモリ | 11,392 KB |
最終ジャッジ日時 | 2024-07-22 07:35:01 |
合計ジャッジ時間 | 16,592 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
use std::collections::{VecDeque, HashSet}; const INF: usize = 1usize << 60; fn bfs(start: usize, paths: &Vec<Vec<usize>>) -> Vec<usize> { let mut ret = vec![INF; paths.len()]; ret[start] = 0usize; let mut deque = VecDeque::new(); deque.push_back(start); while let Some(u) = deque.pop_front() { for &v in paths[u].iter() { if ret[v] < INF { continue; } ret[v] = ret[u] + 1; deque.push_back(v); } } ret } 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 a = temp[0]-1; let b = temp[1]-1; paths[a].push(b); paths[b].push(a); } let n3 = paths.iter().filter(|&v| v.len() > 2).count(); if n3 > 1 { println!("No"); } else if n3 == 0 { println!("Yes"); } else { let idx = (0..n).filter(|&i| paths[i].len() > 2).nth(0).unwrap(); let dists1 = bfs(idx, &paths); let dist_pattern = (0..n).filter(|&i| paths[i].len() == 1).map(|i| dists1[i]).collect::<HashSet<usize>>().len(); if dist_pattern == 1 && paths.iter().filter(|&v| v.len() > 2).count() <= 1 { println!("Yes"); } else { println!("No"); } } }