結果
問題 | No.2563 色ごとのグループ |
ユーザー | Spica |
提出日時 | 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 |
ソースコード
#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; }