結果

問題 No.1293 2種類の道路
ユーザー 👑 Nachia
提出日時 2020-11-20 22:42:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 149 ms / 2,000 ms
コード長 865 bytes
コンパイル時間 2,251 ms
コンパイル使用メモリ 204,596 KB
最終ジャッジ日時 2025-01-16 02:46:55
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
using namespace std;
using LL=long long;
using ULL=unsigned long long;
#define rep(i,n) for(int i=0;i<(n);i++)

struct dsu{
  vector<int> V;
  vector<int> Z;
  dsu(int n){ V.resize(n); rep(i,n) V[i]=i; Z.assign(n,1); }
  int root(int a){ if(V[a]==a) return a; return V[a]=root(V[a]); }
  int size(int a){ return Z[root(a)]; }
  void merge(int u,int v){ u=root(u); v=root(v); V[v]=u; if(u!=v)Z[u]+=Z[v]; }
};

int main(){
  int N,D,W; cin>>N>>D>>W;
  dsu A(N), B(N);
  set<pair<int,int>> S;
  rep(i,D){
    int u,v; cin>>u>>v; u--; v--;
    A.merge(u,v);
  }
  rep(i,W){
    int u,v; cin>>u>>v; u--; v--;
    B.merge(u,v);
  }

  rep(i,N){
    S.insert({A.root(i),B.root(i)});
  }

  LL ans=0;
  for(pair<int,int> p:S){
    LL zA = A.size(p.first);
    LL zB = B.size(p.second);
    ans += zA * zB;
  }
  cout<<(ans-N)<<endl;

  return 0;
}

0