結果
問題 | No.1390 Get together |
ユーザー | akakimidori |
提出日時 | 2021-02-12 21:30:10 |
言語 | Rust (1.72.1) |
結果 |
AC
|
実行時間 | 120 ms / 2,000 ms |
コード長 | 3,558 bytes |
コンパイル時間 | 5,279 ms |
コンパイル使用メモリ | 158,488 KB |
実行使用メモリ | 21,568 KB |
最終ジャッジ日時 | 2023-09-27 02:57:20 |
合計ジャッジ時間 | 8,948 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge11 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,380 KB |
testcase_03 | AC | 2 ms
4,376 KB |
testcase_04 | AC | 2 ms
4,376 KB |
testcase_05 | AC | 2 ms
4,380 KB |
testcase_06 | AC | 2 ms
4,380 KB |
testcase_07 | AC | 2 ms
4,376 KB |
testcase_08 | AC | 2 ms
4,380 KB |
testcase_09 | AC | 2 ms
4,376 KB |
testcase_10 | AC | 1 ms
4,380 KB |
testcase_11 | AC | 1 ms
4,380 KB |
testcase_12 | AC | 1 ms
4,376 KB |
testcase_13 | AC | 1 ms
4,380 KB |
testcase_14 | AC | 1 ms
4,376 KB |
testcase_15 | AC | 1 ms
4,380 KB |
testcase_16 | AC | 24 ms
10,892 KB |
testcase_17 | AC | 110 ms
21,408 KB |
testcase_18 | AC | 25 ms
10,844 KB |
testcase_19 | AC | 119 ms
21,512 KB |
testcase_20 | AC | 120 ms
21,568 KB |
testcase_21 | AC | 117 ms
21,456 KB |
testcase_22 | AC | 70 ms
17,580 KB |
testcase_23 | AC | 71 ms
17,572 KB |
testcase_24 | AC | 69 ms
17,660 KB |
testcase_25 | AC | 117 ms
21,408 KB |
testcase_26 | AC | 120 ms
21,480 KB |
testcase_27 | AC | 118 ms
21,480 KB |
testcase_28 | AC | 117 ms
21,516 KB |
testcase_29 | AC | 116 ms
21,480 KB |
testcase_30 | AC | 115 ms
21,512 KB |
testcase_31 | AC | 117 ms
21,496 KB |
ソースコード
//---------- begin union_find ---------- pub struct DSU { p: Vec<i32>, } impl DSU { pub fn new(n: usize) -> DSU { assert!(n < std::i32::MAX as usize); DSU { p: vec![-1; n] } } pub fn init(&mut self) { self.p.iter_mut().for_each(|p| *p = -1); } pub fn root(&self, mut x: usize) -> usize { assert!(x < self.p.len()); while self.p[x] >= 0 { x = self.p[x] as usize; } x } pub fn same(&self, x: usize, y: usize) -> bool { assert!(x < self.p.len() && y < self.p.len()); self.root(x) == self.root(y) } pub fn unite(&mut self, x: usize, y: usize) -> Option<(usize, usize)> { assert!(x < self.p.len() && y < self.p.len()); let mut x = self.root(x); let mut y = self.root(y); if x == y { return None; } if self.p[x] > self.p[y] { std::mem::swap(&mut x, &mut y); } self.p[x] += self.p[y]; self.p[y] = x as i32; Some((x, y)) } pub fn parent(&self, x: usize) -> Option<usize> { assert!(x < self.p.len()); let p = self.p[x]; if p >= 0 { Some(p as usize) } else { None } } pub fn sum<F>(&self, mut x: usize, mut f: F) -> usize where F: FnMut(usize), { while let Some(p) = self.parent(x) { f(x); x = p; } x } pub fn size(&self, x: usize) -> usize { assert!(x < self.p.len()); let r = self.root(x); (-self.p[r]) as usize } } //---------- end union_find ---------- // ---------- begin input macro ---------- // reference: https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8 macro_rules! input { (source = $s:expr, $($r:tt)*) => { let mut iter = $s.split_whitespace(); input_inner!{iter, $($r)*} }; ($($r:tt)*) => { let s = { use std::io::Read; let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); s }; let mut iter = s.split_whitespace(); input_inner!{iter, $($r)*} }; } macro_rules! input_inner { ($iter:expr) => {}; ($iter:expr, ) => {}; ($iter:expr, $var:ident : $t:tt $($r:tt)*) => { let $var = read_value!($iter, $t); input_inner!{$iter $($r)*} }; } macro_rules! read_value { ($iter:expr, ( $($t:tt),* )) => { ( $(read_value!($iter, $t)),* ) }; ($iter:expr, [ $t:tt ; $len:expr ]) => { (0..$len).map(|_| read_value!($iter, $t)).collect::<Vec<_>>() }; ($iter:expr, chars) => { read_value!($iter, String).chars().collect::<Vec<char>>() }; ($iter:expr, bytes) => { read_value!($iter, String).bytes().collect::<Vec<u8>>() }; ($iter:expr, usize1) => { read_value!($iter, usize) - 1 }; ($iter:expr, $t:ty) => { $iter.next().unwrap().parse::<$t>().expect("Parse error") }; } // ---------- end input macro ---------- fn run() { input! { n: usize, m: usize, p: [(usize1, usize1); n], } let mut map = std::collections::BTreeMap::new(); for (b, c) in p { map.entry(c).or_insert(vec![]).push(b); } let mut dsu = DSU::new(m); let mut ans = 0; for (_, b) in map { for b in b.windows(2) { if dsu.unite(b[0], b[1]).is_some() { ans += 1; } } } println!("{}", ans); } fn main() { run(); }