結果

問題 No.2290 UnUnion Find
ユーザー Sarievo
提出日時 2023-05-05 23:02:16
言語 Rust
(1.83.0 + proconio)
結果
RE  
実行時間 -
コード長 3,690 bytes
コンパイル時間 15,783 ms
コンパイル使用メモリ 379,664 KB
実行使用メモリ 38,784 KB
最終ジャッジ日時 2024-11-23 11:17:04
合計ジャッジ時間 35,070 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20 WA * 22 RE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

pub struct DisjointSetUnion {
    parent: Vec<usize>,
    sizes: Vec<usize>,
    size: usize,
}
impl DisjointSetUnion {
    pub fn new(n: usize) -> DisjointSetUnion {
        DisjointSetUnion {
            parent: (0..n).map(|i| i).collect::<Vec<usize>>(),
            sizes: vec![1; n],
            size: n,
        }
    }
    pub fn leader(&mut self, x: usize) -> usize {
        if x == self.parent[x] { return x; }
        let px = self.parent[x];
        self.parent[x] = self.leader(px);
        self.parent[x]
    }
    pub fn merge(&mut self, x: usize, y: usize) -> bool {
        let px = self.leader(x);
        let py = self.leader(y);
        if px == py { return false; }
        let (l, s) = if self.sizes[px] < self.sizes[py] { (py, px) } else { (px, py) };
        self.parent[s] = l;
        self.sizes[l] += self.sizes[s];
        self.sizes[s] = 0;
        self.size -= 1;
        true
    }
}
use std::collections::{BTreeMap, HashSet};
use DisjointSetUnion as DSU;
fn main() {
    let (r, w) = (std::io::stdin(), std::io::stdout());
    let mut sc = IO::new(r.lock(), w.lock());
    let n = sc.read();
    let q = sc.read();
    let mut dsu = DSU::new(n);
    let mut sets: BTreeMap<usize, HashSet<usize>> = BTreeMap::new();
    for i in 0..n {
        let mut set = HashSet::new();
        set.insert(i);
        sets.insert(i, set);
    }
    for _ in 0..q {
        let t = sc.read();
        match t {
            1 => {
                let u = sc.usize0();
                let v = sc.usize0();
                let pu = dsu.leader(u);
                let pv = dsu.leader(v);
                if !dsu.merge(u, v) { continue; }
                let p = dsu.leader(u);
                sets.get_mut(&pu).unwrap().remove(&u);
                sets.get_mut(&pv).unwrap().remove(&v);
                sets.get_mut(&p).unwrap().insert(u);
                sets.get_mut(&p).unwrap().insert(v);
                if sets.get_mut(&pu).unwrap().is_empty() { sets.remove(&pu); }
                if sets.get_mut(&pv).unwrap().is_empty() { sets.remove(&pv); }
            }
            2 => {
                let u = sc.usize0();
                let size = dsu.sizes[u];
                if size == n {
                    println!("{}", -1);
                } else {
                    let id = dsu.leader(u);
                    let mut set = sets.range(id + 1..).next();
                    if set.is_none() { set = sets.range(..id).next_back(); }
                    println!("{}", set.unwrap().1.iter().nth(0).unwrap() + 1);
                }
            }
            _ => unreachable!()
        }
    }
}
pub struct IO<R, W: std::io::Write>(R, std::io::BufWriter<W>);
impl<R: std::io::Read, W: std::io::Write> IO<R, W> {
    pub fn new(r: R, w: W) -> Self { Self(r, std::io::BufWriter::new(w)) }
    pub fn write<S: ToString>(&mut self, s: S) {
        use std::io::Write;
        self.1.write_all(s.to_string().as_bytes()).unwrap();
    }
    pub fn read<T: std::str::FromStr>(&mut self) -> T {
        use std::io::Read;
        let buf = self.0.by_ref().bytes()
            .map(|b| b.unwrap())
            .skip_while(|&b| b == b' ' || b == b'\n' || b == b'\r' || b == b'\t')
            .take_while(|&b| b != b' ' && b != b'\n' && b != b'\r' && b != b'\t')
            .collect::<Vec<_>>();
        unsafe { std::str::from_utf8_unchecked(&buf) }.parse().ok().expect("Parse error.")
    }
    pub fn chars(&mut self) -> Vec<char> { self.read::<String>().chars().collect() }
    pub fn usize0(&mut self) -> usize { self.read::<usize>() - 1 }
    pub fn vec<T: std::str::FromStr>(&mut self, n: usize) -> Vec<T> {
        (0..n).map(|_| self.read()).collect()
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0