結果

問題 No.2563 色ごとのグループ
ユーザー Feishi LiFeishi Li
提出日時 2023-12-02 15:57:34
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 132 ms / 2,000 ms
コード長 2,005 bytes
コンパイル時間 4,357 ms
コンパイル使用メモリ 267,916 KB
実行使用メモリ 24,448 KB
最終ジャッジ日時 2024-09-26 19:35:05
合計ジャッジ時間 7,470 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 AC 1 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 1 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 1 ms
5,376 KB
testcase_13 AC 1 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 3 ms
5,376 KB
testcase_17 AC 2 ms
5,376 KB
testcase_18 AC 2 ms
5,376 KB
testcase_19 AC 4 ms
5,376 KB
testcase_20 AC 9 ms
5,376 KB
testcase_21 AC 6 ms
5,376 KB
testcase_22 AC 4 ms
5,376 KB
testcase_23 AC 7 ms
5,376 KB
testcase_24 AC 64 ms
12,160 KB
testcase_25 AC 54 ms
12,800 KB
testcase_26 AC 122 ms
17,532 KB
testcase_27 AC 120 ms
20,864 KB
testcase_28 AC 97 ms
19,088 KB
testcase_29 AC 122 ms
24,448 KB
testcase_30 AC 123 ms
24,448 KB
testcase_31 AC 124 ms
24,448 KB
testcase_32 AC 124 ms
24,448 KB
testcase_33 AC 116 ms
21,496 KB
testcase_34 AC 125 ms
21,620 KB
testcase_35 AC 132 ms
21,500 KB
testcase_36 AC 121 ms
21,624 KB
testcase_37 AC 125 ms
21,500 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <bits/extc++.h>
using i64 = long long;

struct Dsu
{
    int n;
    std::vector<int> f, sz;

    Dsu() {}
    Dsu(int _n) : n(_n)
    {
        f.resize(n);
        std::iota(f.begin(), f.end(), 0);
        sz.assign(n, 1);
    }
    int leader(int x)
    {
        if (f[x] == x)
            return x;
        return f[x] = leader(f[x]);
    }
    void merge(int u, int v)
    {
        u = leader(u), v = leader(v);
        if (u == v)
            return;
        if (sz[u] < sz[v])
        {
            std::swap(u, v);
        }
        f[v] = u;
        sz[u] += sz[v];
    }
    bool same(int u, int v)
    {
        return leader(u) == leader(v);
    }
    int size(int x)
    {
        return sz[leader(x)];
    }
};

void solve()
{
    int n, m;
    std::cin >> n >> m;
    std::vector<int> color(n);
    int ans = 0;
    for (auto &i : color)
    {
        std::cin >> i;
        i--;
    }

    std::vector<std::vector<int>> adj(n);
    for (int i = 0; i < m; i++)
    {
        int u, v;
        std::cin >> u >> v;
        --u, --v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    Dsu dsu(n);

    for (int i = 0; i < n; i++)
    {
        if (adj[i].empty())
            continue;
        for (const auto &j : adj[i])
        {
            if (color[i] == color[j])
            {
                dsu.merge(i, j);
            }
        }
    }

    std::vector<std::vector<int>> id(n);
    for (int i = 0; i < n; i++)
    {
        id[color[i]].push_back(i);
    }

    for (const auto &v : id)
    {
        if (v.size() <= 1)
        {
            continue;
        }
        int x = v.front();
        for (size_t i = 1; i < v.size(); i++)
        {
            int y = v[i];
            if (!dsu.same(x, y))
            {
                ans++;
                dsu.merge(x, y);
            }
        }
    }
    std::cout << ans << '\n';
}
int main()
{
    std::cin.tie(nullptr)->sync_with_stdio(false);
    solve();
    return 0;
}
0