結果
問題 | No.1767 BLUE to RED |
ユーザー | phspls |
提出日時 | 2022-10-15 16:11:03 |
言語 | Rust (1.77.0 + proconio) |
結果 |
TLE
|
実行時間 | - |
コード長 | 3,248 bytes |
コンパイル時間 | 15,848 ms |
コンパイル使用メモリ | 387,896 KB |
実行使用メモリ | 25,020 KB |
最終ジャッジ日時 | 2024-06-26 19:57:32 |
合計ジャッジ時間 | 19,214 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
13,752 KB |
testcase_01 | AC | 1 ms
6,812 KB |
testcase_02 | AC | 1 ms
6,940 KB |
testcase_03 | AC | 1 ms
6,940 KB |
testcase_04 | AC | 3 ms
6,940 KB |
testcase_05 | AC | 4 ms
6,940 KB |
testcase_06 | AC | 1 ms
6,940 KB |
testcase_07 | AC | 6 ms
6,940 KB |
testcase_08 | AC | 9 ms
6,944 KB |
testcase_09 | TLE | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
コンパイルメッセージ
warning: field `n` is never read --> src/main.rs:4:5 | 3 | struct UnionFind { | --------- field in this struct 4 | n: usize, | ^ | = note: `#[warn(dead_code)]` on by default warning: method `chain_cnt` is never used --> src/main.rs:47:8 | 10 | impl UnionFind { | -------------- method in this implementation ... 47 | fn chain_cnt(&mut self, i: usize) -> usize { | ^^^^^^^^^
ソースコード
use std::{collections::{BTreeSet, HashMap, BinaryHeap}, cmp::Reverse}; 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 mut a = String::new(); std::io::stdin().read_line(&mut a).ok(); let a: BTreeSet<usize> = a.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let mut b = String::new(); std::io::stdin().read_line(&mut b).ok(); let b: BTreeSet<usize> = b.trim().split_whitespace().map(|s| s.parse().unwrap()).collect(); let mut uf = UnionFind::new(n+m+1); let mut totals = BTreeSet::new(); for &v in a.iter() { totals.insert(v); } for &v in b.iter() { totals.insert(v); } let mapping = (0..n+m).map(|i| (*totals.iter().nth(i).unwrap(), i)).collect::<HashMap<usize, usize>>(); for &v in a.iter() { uf.unite(*mapping.get(&v).unwrap(), n+m); } let mut pque = BinaryHeap::new(); for &v in b.iter() { let idx = *mapping.get(&v).unwrap(); if let Some(&left) = a.range(..v).rev().nth(0) { let nidx = *mapping.get(&left).unwrap(); pque.push((Reverse(v-left), idx, nidx)); } if let Some(&right) = a.range(v+1..).nth(0) { let nidx = *mapping.get(&right).unwrap(); pque.push((Reverse(right-v), idx, nidx)); } if let Some(&left) = b.range(..v).rev().nth(0) { let nidx = *mapping.get(&left).unwrap(); pque.push((Reverse(v-left), idx, nidx)); } if let Some(&right) = b.range(v+1..).nth(0) { let nidx = *mapping.get(&right).unwrap(); pque.push((Reverse(right-v), idx, nidx)); } } let mut result = 0usize; while let Some((Reverse(cost), l, r)) = pque.pop() { if uf.find(l) == uf.find(r) { continue; } uf.unite(l, r); result += cost; } println!("{}", result); }