結果

問題 No.2563 色ごとのグループ
ユーザー lemon_math_tea
提出日時 2023-12-02 15:30:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 394 ms / 2,000 ms
コード長 2,717 bytes
コンパイル時間 4,454 ms
コンパイル使用メモリ 263,284 KB
最終ジャッジ日時 2025-02-18 04:40:18
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff #

// コンパイルはctrl + shift + B
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
#define ll long long
#define all(x) (x).begin(), (x).end()
#define yes "Yes"
#define no "No"

// 最大公約数 O(log(N))
template<typename T>
T gcd(T a, T b) {
    if (b == 0) return a;
    else return gcd(b, a % b);
}

// 素数判定 O(sqrt(N))
bool is_prime(ll n) {
    for (ll i = 2; i*i <= n; i++) {
        if (n % i == 0) return false;
    }
    return true;
}

//n以下の素数求めマシーン O(Nloglog(N))
vector<ll> getprime(ll n) {
    vector<ll> ans;
    vector<bool> isprime(n + 1, true);
    //0, 1はダメ
    isprime[0] = false; isprime[1] = false;

    //やる
    for (int i = 2; i <= n; i++) {
        if (!isprime[i]) continue;
        ans.push_back(i);

        for (int j =  2 * i; j <= n; j += i) {
            //i以外のiの倍数をキャンセル
            isprime[j] = false;
        }
    }
    return ans;
}

// nを素因数分解 O(sqrt(N))
vector<pair<ll, ll>> prime_factorize(ll n) {
    vector<pair<ll, ll>> ans;
    for (ll i = 2; i*i <= n; i++) {
        if (n % i != 0) continue;
        ll e = 0;
        while (n % i == 0) {
            e++;
            n /= i;
        }
        ans.push_back({i, e});
    }
    if (n != 1) ans.push_back({n, 1});
    return ans;
}

// 各n以下の最小の素因数 O(NlogN)
vector<ll> smallest_prime(ll n) {
    vector<ll> spf(n+1);
    for (ll i = 0; i <= n; i++) spf[i] = i; //初期化
    for (ll i = 2; i+i <= n; i++) {
        if (spf[i] == i) {
            for (ll j = i*i; j <= n; j += i) {
                if (spf[j] == j) spf[j] = i;
            }
        }
    }
    return spf;
}

// 以下、コードを記すのである。

// using mint = modint998244353;
using mint = modint1000000007;

int main() {
    int n, m; cin >> n >> m;
    vector<int> c(n);
    for (int i = 0; i < n; i++) cin >> c[i];
    vector<vector<int>> e(n);
    for (int i = 0; i < m; i++) {
        int u, v; cin >> u >> v;
        u--; v--;
        e[u].push_back(v);
        e[v].push_back(u);
    }
    map<int, int> mp;
    vector<bool> seen(n);
    queue<int> q;
    for (int i = 0; i < n; i++) {
        if (seen[i]) continue;
        seen[i] = true;
        q.push(i);
        mp[c[i]]++;
        while (q.size()) {
            int u = q.front(); q.pop();
            int col = c[u];
            for (auto v : e[u]) {
                if (col != c[v]) continue;
                if (seen[v]) continue;
                seen[v] = true;
                q.push(v);

            }
        }
    }
    int ans = 0;
    for (auto p : mp) {
        ans += p.second - 1;
    }
    cout << ans << endl;
}
0