結果

問題 No.2563 色ごとのグループ
コンテスト
ユーザー GOTKAKO
提出日時 2023-12-02 14:57:43
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 142 ms / 2,000 ms
コード長 1,453 bytes
コンパイル時間 2,754 ms
コンパイル使用メモリ 206,636 KB
最終ジャッジ日時 2025-02-18 04:07:02
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

class UnionFind{
    public:
    vector<int> par,siz;
 
    void make(int N){
        par.resize(N,-1);
        siz.resize(N,1);
    }
 
    int root(int x){
        if(par.at(x) == -1) return x;
        else return par.at(x) = root(par.at(x));
    }
 
    void unite(int u, int v){
        u = root(u),v = root(v);
        if(u == v) return;
        if(siz.at(u) < siz.at(v)) swap(u,v);
 
        par.at(v) = u;
        siz.at(u) += siz.at(v);
    }
 
    bool issame(int u, int v){
        if(root(u) == root(v)) return true;
        else return false;
    }
    //ノーマルUnion-Findです UnionbySizeと経路圧縮搭載!.
};

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

        int N,M; cin >> N >> M;
        UnionFind Z; Z.make(N);
        vector<int> C(N);
        vector<vector<int>> rev(N);
        for(int i=0; i<N; i++){
            int c; cin >> c; c--;
            C.at(i) = c;
            rev.at(c).push_back(i);
        }

        for(int i=0; i<M; i++){
            int u,v; cin >> u >> v;
            u--; v--;
            if(C.at(u) != C.at(v)) continue;
            Z.unite(u,v);
        }

        int answer = 0;
        for(int i=0; i<N; i++){
        	if(rev.at(i).size() == 0) continue;
            set<int> Vs;
            for(auto v : rev.at(i)) Vs.insert(Z.root(v));
            answer += Vs.size()-1;
        }
        cout << answer << endl;
}
0