結果
問題 | No.1779 Magical Swap |
ユーザー | akakimidori |
提出日時 | 2021-12-08 02:00:07 |
言語 | Rust (1.77.0) |
結果 |
AC
|
実行時間 | 56 ms / 2,000 ms |
コード長 | 2,581 bytes |
コンパイル時間 | 1,434 ms |
コンパイル使用メモリ | 156,696 KB |
実行使用メモリ | 13,836 KB |
最終ジャッジ日時 | 2023-09-23 07:48:46 |
合計ジャッジ時間 | 3,105 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 3 ms
4,376 KB |
testcase_03 | AC | 2 ms
4,380 KB |
testcase_04 | AC | 52 ms
11,160 KB |
testcase_05 | AC | 22 ms
5,852 KB |
testcase_06 | AC | 24 ms
6,376 KB |
testcase_07 | AC | 44 ms
9,800 KB |
testcase_08 | AC | 2 ms
4,376 KB |
testcase_09 | AC | 13 ms
4,376 KB |
testcase_10 | AC | 54 ms
11,728 KB |
testcase_11 | AC | 27 ms
6,924 KB |
testcase_12 | AC | 10 ms
4,380 KB |
testcase_13 | AC | 42 ms
8,996 KB |
testcase_14 | AC | 56 ms
11,812 KB |
testcase_15 | AC | 38 ms
13,836 KB |
testcase_16 | AC | 56 ms
11,756 KB |
testcase_17 | AC | 32 ms
4,696 KB |
testcase_18 | AC | 1 ms
4,376 KB |
ソースコード
// ---------- begin quick find ---------- pub struct QuickFind { size: usize, id: Vec<usize>, list: Vec<Vec<usize>>, } impl QuickFind { pub fn new(size: usize) -> Self { let id = (0..size).collect::<Vec<_>>(); let list = (0..size).map(|x| vec![x]).collect::<Vec<_>>(); QuickFind { size, id, list } } pub fn root(&self, x: usize) -> usize { assert!(x < self.size); self.id[x] } pub fn same(&self, x: usize, y: usize) -> bool { assert!(x < self.size); assert!(y < self.size); self.root(x) == self.root(y) } pub fn unite(&mut self, x: usize, y: usize) -> Option<(usize, usize)> { assert!(x < self.size); assert!(y < self.size); let mut x = self.root(x); let mut y = self.root(y); if x == y { return None; } if (self.list[x].len(), x) < (self.list[y].len(), y) { std::mem::swap(&mut x, &mut y); } let mut z = std::mem::take(&mut self.list[y]); z.iter().for_each(|y| self.id[*y] = x); self.list[x].append(&mut z); Some((x, y)) } pub fn enumerate(&self, x: usize) -> &[usize] { assert!(x < self.size); &self.list[self.root(x)] } pub fn size(&self, x: usize) -> usize { assert!(x < self.size); self.enumerate(x).len() } } // ---------- end quick find ---------- use std::io::*; fn read() -> Vec<(Vec<u32>, Vec<u32>)> { let mut s = String::new(); std::io::stdin().read_to_string(&mut s).unwrap(); let mut it = s.trim().split_whitespace().flat_map(|s| s.parse::<u32>()); let mut next = || it.next().unwrap(); let t = next(); (0..t).map(|_| { let n = next(); let a = (0..n).map(|_| next()).collect(); let b = (0..n).map(|_| next()).collect(); (a, b) }).collect() } fn main() { let out = std::io::stdout(); let mut out = std::io::BufWriter::new(out.lock()); for (a, b) in read() { let n = a.len(); let mut dsu = QuickFind::new(n); for k in 2..=n { for j in 2..=(n / k) { dsu.unite(k * (j - 1) - 1, k * j - 1); } } let mut a = a.iter().enumerate().map(|a| (dsu.root(a.0), *a.1)).collect::<Vec<_>>(); let mut b = b.iter().enumerate().map(|a| (dsu.root(a.0), *a.1)).collect::<Vec<_>>(); a.sort(); b.sort(); if a == b { writeln!(out, "Yes").ok(); } else { writeln!(out, "No").ok(); } } }