結果

問題 No.2563 色ごとのグループ
ユーザー SpicaSpica
提出日時 2023-12-02 15:16:10
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 213 ms / 2,000 ms
コード長 1,993 bytes
コンパイル時間 4,902 ms
コンパイル使用メモリ 268,000 KB
実行使用メモリ 13,600 KB
最終ジャッジ日時 2024-09-26 18:21:15
合計ジャッジ時間 7,906 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include<atcoder/all>
using namespace std;
using namespace atcoder;
using ll = long long; // -2^63 ~ 2^63-1 (9.2*10^18)
using ull = unsigned long long; // 0 ~ 2^64-1 (1.8*10^19)
using ld = long double;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rep2(i, m, n) for (int i = (m); (i) < int (n); i++)
#define all(v) v.begin(), v.end()
#define bit(n) (1ll<<(n)) // 2^n
#define sz(x) ((int)(x).size())
#define fi first
#define se second
template<class T> using maxpq = priority_queue<T>;
template<class T> using minpq = priority_queue<T, vector<T>, greater<T>>;
template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }
template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }
const ll llINF = 1000000000000000000;
void YesNo(bool flag){
    if(flag) cout << "Yes" << endl;
    else cout << "No" << endl;
}
// 多次元vector生成
template<ll idx = 0, ll n, class T>
auto make_vec(const ll (&d)[n], T&& init) noexcept {
    if constexpr (idx < n) return std::vector(d[idx], make_vec<idx + 1>(d, std::forward<T>(init)));
    else return init;
}
template<class T, ll idx = 0, ll n>
auto make_vec(const ll (&d)[n], const T& init = {}) noexcept {
    if constexpr (idx < n) return std::vector(d[idx], make_vec<idx + 1>(d, init));
    else return init;
}
// auto 変数名 = make_vec<型>({a, b, ...}, 初期値);

int main(){
    int N, M;
    cin >> N >> M;
    vector<int> C(N);
    vector<vector<int>> group(N);
    rep(i, N){
        cin >> C[i];
        C[i]--;
        group[C[i]].push_back(i);
    }

    dsu uf(N);
    rep(i, M){
        int u, v;
        cin >> u >> v;
        u--, v--;
        if(C[u] == C[v]){
            uf.merge(u, v);
        }
    }

    int ans = 0;
    rep(i, N){
        if(sz(group[i]) == 0) continue;
        set<int> s;
        for(auto val : group[i]){
            s.insert(uf.leader(val));
        }
        ans += sz(s)-1;
    }
    cout << ans << endl;
}
0