結果
問題 |
No.1390 Get together
|
ユーザー |
|
提出日時 | 2022-11-21 00:20:57 |
言語 | Rust (1.83.0 + proconio) |
結果 |
AC
|
実行時間 | 156 ms / 2,000 ms |
コード長 | 2,288 bytes |
コンパイル時間 | 15,947 ms |
コンパイル使用メモリ | 385,244 KB |
実行使用メモリ | 28,160 KB |
最終ジャッジ日時 | 2024-09-21 20:32:11 |
合計ジャッジ時間 | 21,681 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
コンパイルメッセージ
warning: field `n` is never read --> src/main.rs:5:5 | 4 | struct UnionFind { | --------- field in this struct 5 | n: usize, | ^ | = note: `#[warn(dead_code)]` on by default warning: method `chain_cnt` is never used --> src/main.rs:48:8 | 11 | impl UnionFind { | -------------- method in this implementation ... 48 | fn chain_cnt(&mut self, i: usize) -> usize { | ^^^^^^^^^
ソースコード
use std::collections::{HashSet, HashMap}; struct UnionFind { n: usize, parents: Vec<usize>, depths: Vec<usize>, chains: Vec<usize>, } impl UnionFind { fn new(n: usize) -> Self { UnionFind { n: n, parents: (0..n).collect(), depths: vec![0; n], chains: vec![1; n], } } fn equiv(&mut self, a: usize, b: usize) -> bool { self.find(a) == self.find(b) } fn unite(&mut self, a: usize, b: usize) { if self.equiv(a, b) { return; } let x = self.parents[a]; let y = self.parents[b]; if self.depths[x] > self.depths[y] { self.parents[x] = self.parents[y]; self.chains[y] += self.chains[x]; } else { self.parents[y] = self.parents[x]; self.chains[x] += self.chains[y]; if self.depths[x] == self.depths[y] { self.depths[x] += 1; } } } fn find(&mut self, a: usize) -> usize { if self.parents[a] == a { return a; } let p = self.find(self.parents[a]); self.parents[a] = p; p } fn chain_cnt(&mut self, i: usize) -> usize { let idx = self.find(i); self.chains[idx] } } fn main() { let mut nm = String::new(); std::io::stdin().read_line(&mut nm).ok(); let nm: Vec<usize> = nm.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let n = nm[0]; let m = nm[1]; let slimes = (0..n).map(|_| { 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(); (temp[0]-1, temp[1]-1) }) .collect::<Vec<_>>(); let mut color_box = vec![vec![]; m]; for &(b, c) in slimes.iter() { color_box[b].push(c); } let mut uf = UnionFind::new(n); for c in color_box.iter() { for &v in c.iter() { uf.unite(c[0], v); } } let mut mapping = HashMap::new(); for &(b, c) in slimes.iter() { mapping.entry(uf.find(c)).or_insert(HashSet::new()).insert(b); } println!("{}", mapping.values().map(|v| v.len()-1).sum::<usize>()); }