結果

問題 No.2563 色ごとのグループ
ユーザー int_san
提出日時 2023-12-02 15:30:33
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 291 ms / 2,000 ms
コード長 1,326 bytes
コンパイル時間 1,844 ms
コンパイル使用メモリ 178,888 KB
実行使用メモリ 25,216 KB
最終ジャッジ日時 2024-09-26 18:48:54
合計ジャッジ時間 6,448 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
struct UnionFind {
    vector<int> parent, rank, siz;
    UnionFind(int n) {
        parent.resize(n, -1);
        rank.resize(n, 1);
        siz.resize(n, 1);
    }
    int find(int x) {
        if (parent[x] == -1) {
            return x;
        } else {
            return parent[x] = find(parent[x]);
        }
    }
    void unite(int x, int y) {
        x = find(x);
        y = find(y);
        if (x == y) {
            return;
        }
        if (rank[x] < rank[y]) {
            swap(x, y);
        }
        parent[y] = x;
        if (rank[x] == rank[y]) {
            rank[x]++;
        }
        siz[x] += siz[y];
    }
    bool same(int x, int y) { return find(x) == find(y); }
    int size(int x) { return siz[find(x)]; }
};
int main() {
    int N, M;
    cin >> N >> M;
    vector<int> C(N);
    for (int i = 0; i < N; i++) {
        cin >> C[i];
        C[i]--;
    }
    UnionFind uf(N);
    for (int i = 0; i < M; i++) {
        int u, v;
        cin >> u >> v;
        u--;
        v--;
        if (C[u] == C[v]) uf.unite(u, v);
    }
    int ans = 0;
    vector<set<int>> g(N);
    for (int i = 0; i < N; i++) {
        g[C[i]].insert(uf.find(i));
    }
    for (auto i : g) {
        if (i.size() > 0) ans += i.size() - 1;
    }
    cout << ans << endl;
}
0