結果

問題 No.2563 色ごとのグループ
コンテスト
ユーザー cumene_rx
提出日時 2023-12-02 15:25:51
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 464 ms / 2,000 ms
コード長 4,681 bytes
コンパイル時間 5,484 ms
コンパイル使用メモリ 328,812 KB
実行使用メモリ 37,576 KB
最終ジャッジ日時 2024-09-26 18:40:53
合計ジャッジ時間 11,086 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
#include <random>
using namespace std;
using namespace atcoder;
typedef long long ll;
typedef unsigned long long ull;
using mint9 = modint998244353;
using mint1 = modint1000000007;
//using mint = modint;
const ll INF = 1000000007;
const ll LINF = INF * INF;
//const long long LLMAX = 9223372036854775807;
const long long LLMAX = 100000000000000000;

const double TIME_1 = 1.5;
//const ll MOD = 998244353;
std::mt19937 get_rand_mt;
#define PA pair<int,int>
#define PAL pair<ll,ll>
#define TP tuple<int,int,int>
#define TPL tuple<ll, ll, ll>
#define rep(i, z, n) for(int i = z; i < n; i++)
#define repl(i, z, n) for(ll i = z; i < n; i++)
#define revrep(i, z, n) for(int i = n - 1; i >= z; i--)
#define revrepl(i, z, n) for(ll i = n - 1; i >= z; i--)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
#define SZ(x) ((int)(x).size())
#define PB push_back
#define YN(T) if(T){cout<<"Yes"<<endl;}else{cout<<"No"<<endl;}
#define YN2(T) if(T){cout<<"YES"<<endl;}else{cout<<"NO"<<endl;}
#define OK cout << "OK" << endl;
#define FX(x) cout << fixed << setprecision(10) << x << endl;
#define shuffle(x) shuffle(all(x), get_rand_mt)
#define PQ priority_queue<int,vector<int>,greater<int>>
#define PQL priority_queue<ll,vector<ll>,greater<ll>>
#define PQPA priority_queue<PA,vector<PA>,greater<PA>>

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; }
template<class T>T ceilX(T a, T b) {return (a + b - 1)/b; }
ll lmax(ll a, ll b){if(a>b) return a; return b;}
ll lmin(ll a, ll b){if(a>b) return b; return a;}
ll comb(ll X, ll Y){ll res = 1LL;ll R = X;repl(i, 1LL, lmin(Y + 1LL, X - Y + 1LL)){res *= (R - i + 1LL);res /= i;}return res;}
template<class T>void Rekkyo(vector<T> X){cout<<"Rekkyo"<<endl;for(auto Y:X){cout<<Y<<" ";}cout<<endl<<"end"<<endl;}
template<class T>void Rekkyo1(vector<T> X){cout<<"Rekkyo1"<<endl;int cnt=0;for(auto Y:X){cout<<cnt<<" "<<Y<<endl;cnt++;}cout<<"end"<<endl;}
template<class T>void Rekkyo2(vector<vector<T>> X){cout<<"Rekkyo2"<<endl;for(auto Y:X){for(auto Z:Y){cout <<Z<<" ";}cout<<endl;}cout<<"end"<<endl;}

vector<int> Eratosthenes(const int N){
    vector<bool> is_prime(N + 1);
    rep(i, 0, N + 1) is_prime[i] = true;
    vector<int> P;
    rep(i, 2, N + 1){
        if(is_prime[i]){
            for( int j = 2 * i; j <= N; j += i ) is_prime[ j ] = false;
            P.emplace_back(i);
        }
    }
    return P;
}
vector<vector<int>> make_edge(vector<int> u, vector<int> v, bool angle){
    int N = u.size();
    vector<vector<int>> res;
    rep(i, 0, N){res[u[i]].PB(v[i]);if(!angle) res[v[i]].PB(u[i]);}
    return res;
}
vector<vector<PA>> make_edge(vector<int> u, vector<int> v, vector<int> w, bool angle){
    int N = u.size();
    vector<vector<PA>> res;
    rep(i, 0, N){res[u[i]].PB(PA(v[i], w[i]));if(!angle) res[v[i]].PB(PA(u[i], w[i]));}
    return res;
}
vector<vector<PAL>> make_edgeL(vector<int> u, vector<int> v, vector<ll> w, bool angle){
    int N = u.size();
    vector<vector<PAL>> res;
    rep(i, 0, N){res[u[i]].PB(PAL(v[i], w[i]));if(!angle) res[v[i]].PB(PAL(u[i], w[i]));}
    return res;
}









int main(){
    int N, M;
    cin >> N >> M;
    vector<int> C(N);
    vector<vector<int>> nC(N);
    map<int, int> eC;
    set<int> s;
    vector<int> u(M), v(M);
    dsu d(N);
    rep(i, 0, N){
        cin >> C[i];
        C[i]--;
        s.insert(C[i]);
        nC[C[i]].PB(i);
        
        if(eC.count(C[i])){
            eC[C[i]]++;
        }
        else{
            eC[C[i]] = 1;
        }
    }
    rep(i, 0, M){
        cin >> u[i] >> v[i];
        u[i]--;
        v[i]--;
    }
    vector<vector<int>> edge(N);
    map<int, int> edgeCnt;
    rep(m, 0, M){
        //cout << u[m] << " " << v[m] << endl;
        if(C[u[m]] == C[v[m]]){
            //OK;
            if(d.same(u[m], v[m])){
                continue;
            }
            else{
                if(edgeCnt.count(C[u[m]])){
                    edgeCnt[C[u[m]]]++;
                }
                else{
                    edgeCnt[C[u[m]]] = 1;
                }
                d.merge(u[m], v[m]);
            }
        }
    }
    ll ans = 0;
    rep(i, 0, N){
        if(eC[i] <= 1) continue;
        if(eC[i] - 1 == edgeCnt[i]){
            //cout << "OK " << i << " " << eC[i] << " " << edgeCnt[i] << endl;
            continue;
        }
        else{
            //cout << "NG"  << i << " " << eC[i] << " " << edgeCnt[i] << endl;
            ll X = eC[i] - 1 - edgeCnt[i];
            ans += X;
        }
    }
    cout << ans << endl;
    
}
0