結果

問題 No.1293 2種類の道路
ユーザー 👑 Nachia
提出日時 2020-11-20 22:39:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
RE  
実行時間 -
コード長 864 bytes
コンパイル時間 2,256 ms
コンパイル使用メモリ 202,452 KB
最終ジャッジ日時 2025-01-16 02:46:43
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample RE * 2
other RE * 22
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In member function ‘int dsu::merge(int, int)’:
main.cpp:13:77: warning: no return statement in function returning non-void [-Wreturn-type]
   13 |   int merge(int u,int v){ u=root(u); v=root(v); V[v]=u; if(u!=v)Z[u]+=Z[v]; }
      |                                                                             ^

ソースコード

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)]; }
  int 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