結果
問題 | No.2563 色ごとのグループ |
ユーザー |
![]() |
提出日時 | 2023-12-02 15:16:10 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 296 ms / 2,000 ms |
コード長 | 1,993 bytes |
コンパイル時間 | 4,266 ms |
コンパイル使用メモリ | 255,400 KB |
最終ジャッジ日時 | 2025-02-18 04:23:15 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
#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 secondtemplate<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;}