結果

問題 No.2290 UnUnion Find
コンテスト
ユーザー Yukino DX.
提出日時 2024-09-13 22:35:11
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 542 ms / 2,000 ms
コード長 2,964 bytes
コンパイル時間 15,405 ms
コンパイル使用メモリ 378,020 KB
実行使用メモリ 11,264 KB
最終ジャッジ日時 2024-09-13 22:35:47
合計ジャッジ時間 33,108 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::HashSet;

use proconio::{input, marker::Usize1};

fn main() {
    input! {
        n:usize,
        q:usize,
    }

    let mut uf = UnionFindTree::new(n);
    let mut leaders = (0..n).collect::<HashSet<_>>();
    for _ in 0..q {
        input! {
            t:usize,
        }

        if t == 1 {
            input! {
                u:Usize1,
                v:Usize1,
            }

            if !uf.same(u, v) {
                let leader_u = uf.root(u);
                let leader_v = uf.root(v);
                uf.unite(u, v);
                if uf.root(u) == leader_u {
                    leaders.remove(&leader_v);
                } else {
                    leaders.remove(&leader_u);
                }
            }
        } else {
            input! {
                v:Usize1,
            }

            if uf.size(v) == n {
                println!("-1");
            } else {
                let leader_v = uf.root(v);
                let ans = *leaders
                    .iter()
                    .filter(|leader| leader_v != **leader)
                    .next()
                    .unwrap();
                println!("{}", ans + 1);
            }
        }
    }
}

use unionfindtree::*;
mod unionfindtree {
    pub struct UnionFindTree {
        par: Vec<usize>,
        rank: Vec<usize>,
        size: Vec<usize>,
    }

    impl UnionFindTree {
        pub fn new(n: usize) -> Self {
            Self {
                par: (0..n).collect::<Vec<_>>(),
                rank: vec![0; n],
                size: vec![1; n],
            }
        }

        #[allow(dead_code)]
        pub fn same(&mut self, v1: usize, v2: usize) -> bool {
            self.root(v1) == self.root(v2)
        }

        #[allow(dead_code)]
        pub fn size(&mut self, v: usize) -> usize {
            let v_root = self.root(v);
            self.size[v_root]
        }

        #[allow(dead_code)]
        pub fn root(&mut self, v: usize) -> usize {
            if self.par[v] != v {
                self.par[v] = self.root(self.par[v]);
            }

            self.par[v]
        }

        pub fn unite(&mut self, v1: usize, v2: usize) {
            let mut v1_root = self.root(v1);
            let mut v2_root = self.root(v2);
            if v1_root == v2_root {
                return;
            }

            if self.rank[v1_root] < self.rank[v2_root] {
                std::mem::swap(&mut v1_root, &mut v2_root);
            }

            self.par[v2_root] = v1_root;
            self.size[v1_root] += self.size[v2_root];

            if self.rank[v1_root] == self.rank[v2_root] {
                self.rank[v1_root] += 1;
            }
        }

        #[allow(dead_code)]
        pub fn nofcc(&mut self) -> usize {
            let n = self.par.len();
            (0..n)
                .map(|v| self.root(v))
                .collect::<std::collections::HashSet<usize>>()
                .len()
        }
    }
}
0