結果
問題 | No.2563 色ごとのグループ |
ユーザー |
|
提出日時 | 2023-12-02 15:28:46 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 463 ms / 2,000 ms |
コード長 | 1,279 bytes |
コンパイル時間 | 4,427 ms |
コンパイル使用メモリ | 260,812 KB |
最終ジャッジ日時 | 2025-02-18 04:37:50 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using ll = long long; #define MOD 1000000007 #define Mod 998244353 const int MAX = 1000000005; const long long INF = 1000000000000000005LL; using namespace std; using namespace atcoder; int main() { ios::sync_with_stdio(0);cin.tie(); int N, M; cin >> N >> M; vector<int> C(N); for (int i = 0; i < N; i++) {cin >> C[i]; C[i]--;} vector<vector<int>> G(N); for (int i = 0; i < M; i++) { int u, v; cin >> u >> v; u--; v--; if (C[u] != C[v]) continue; G[u].push_back(v); G[v].push_back(u); } vector<bool> visited(N); vector<set<int>> S(N); for (int i = 0; i < N; i++) S[C[i]].insert(i); ll ans = 0; for (int i = 0; i < N; i++) { if (visited[i]) continue; queue<int> que; que.push(i); while (que.size()) { int v = que.front(); que.pop(); S[C[v]].erase(v); visited[v] = true; for (auto &nv : G[v]) { if (!visited[nv]) que.push(nv); } if (que.size() == 0 && S[C[v]].size() > 0) { que.push(*S[C[v]].begin()); ans++; } } } cout << ans << endl; }