結果

問題 No.2912 ０次パーシステントホモロジー
コンテスト
ユーザー Yukino DX.
提出日時 2024-11-06 14:44:10
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 191 ms / 2,000 ms
コード長 2,421 bytes
コンパイル時間 13,704 ms
コンパイル使用メモリ 384,480 KB
実行使用メモリ 12,304 KB
最終ジャッジ日時 2024-11-06 14:44:27
合計ジャッジ時間 16,450 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

use proconio::input;

fn main() {
    input! {
        n:usize,
        m:usize,
        uvw:[(usize,usize,usize);m],
        t:usize,
        r:[usize;t],
    }

    let mut ids_sorted_r = (0..t).collect::<Vec<_>>();
    ids_sorted_r.sort_by_key(|&i| r[i]);

    let mut ids_sorted_w = (0..m).collect::<Vec<_>>();
    ids_sorted_w.sort_by_key(|&i| uvw[i].2);

    let mut ans = vec![0; t];
    let mut uf = UnionFindTree::new(n);

    let mut id = 0;
    for &i in ids_sorted_r.iter() {
        while id < m && uvw[ids_sorted_w[id]].2 <= r[i] {
            let (u, v, _) = uvw[ids_sorted_w[id]];
            if !uf.same(u, v) {
                uf.unite(u, v);
            }
            id += 1;
        }

        ans[i] = uf.ncc();
    }

    for a in ans {
        println!("{}", a);
    }
}

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

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

            self.ncc -= 1;
            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;
            }
        }

        pub fn ncc(&self) -> usize {
            self.ncc
        }
    }
}
0