#934510 (C++23) No.2563 色ごとのグループ

提出ソース

結果

問題	No.2563 色ごとのグループ
ユーザー	cumene_rx
提出日時	2023-12-02 15:25:51
言語	C++23 (gcc 12.3.0 + boost 1.83.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

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,812 KB
testcase_01	AC	2 ms 6,812 KB
testcase_02	AC	2 ms 6,944 KB
testcase_03	AC	1 ms 6,944 KB
testcase_04	AC	2 ms 6,944 KB
testcase_05	AC	2 ms 6,940 KB
testcase_06	AC	2 ms 6,944 KB
testcase_07	AC	1 ms 6,940 KB
testcase_08	AC	2 ms 6,940 KB
testcase_09	AC	2 ms 6,940 KB
testcase_10	AC	2 ms 6,940 KB
testcase_11	AC	2 ms 6,944 KB
testcase_12	AC	2 ms 6,944 KB
testcase_13	AC	2 ms 6,940 KB
testcase_14	AC	4 ms 6,940 KB
testcase_15	AC	4 ms 6,944 KB
testcase_16	AC	4 ms 6,940 KB
testcase_17	AC	3 ms 6,940 KB
testcase_18	AC	3 ms 6,948 KB
testcase_19	AC	6 ms 6,940 KB
testcase_20	AC	26 ms 6,944 KB
testcase_21	AC	14 ms 6,940 KB
testcase_22	AC	10 ms 6,944 KB
testcase_23	AC	17 ms 6,940 KB
testcase_24	AC	186 ms 17,152 KB
testcase_25	AC	181 ms 18,816 KB
testcase_26	AC	287 ms 26,288 KB
testcase_27	AC	347 ms 34,680 KB
testcase_28	AC	314 ms 28,860 KB
testcase_29	AC	438 ms 37,488 KB
testcase_30	AC	454 ms 37,560 KB
testcase_31	AC	450 ms 37,416 KB
testcase_32	AC	464 ms 37,576 KB
testcase_33	AC	236 ms 25,996 KB
testcase_34	AC	224 ms 25,936 KB
testcase_35	AC	223 ms 26,064 KB
testcase_36	AC	228 ms 26,004 KB
testcase_37	AC	239 ms 25,976 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#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;
    
}

yukicoder

結果

テストケース

ソースコード