結果
問題 | No.1293 2種類の道路 |
ユーザー |
![]() |
提出日時 | 2020-11-20 21:33:10 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 209 ms / 2,000 ms |
コード長 | 1,758 bytes |
コンパイル時間 | 1,558 ms |
コンパイル使用メモリ | 138,520 KB |
最終ジャッジ日時 | 2025-01-16 02:07:07 |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #define popcount __builtin_popcount using namespace std; typedef long long int ll; typedef pair<int, int> P; struct unionfind{ vector<int> par, sz; unionfind() {} unionfind(int n):par(n), sz(n, 1){ for(int i=0; i<n; i++) par[i]=i; } int find(int x){ if(par[x]==x) return x; return par[x]=find(par[x]); } void unite(int x, int y){ x=find(x); y=find(y); if(x==y) return; if(sz[x]>sz[y]) swap(x, y); par[x]=y; sz[y]+=sz[x]; } bool same(int x, int y){ return find(x)==find(y); } int size(int x){ return sz[find(x)]; } }; int main() { int n, d, w; cin>>n>>d>>w; unionfind uf1(n), uf2(n); for(int i=0; i<d; i++){ int a, b; cin>>a>>b; a--; b--; uf1.unite(a, b); } for(int i=0; i<w; i++){ int c, d; cin>>c>>d; c--; d--; uf2.unite(c, d); } map<int, vector<int>> mp; for(int i=0; i<n; i++) mp[uf1.find(i)].push_back(i); ll ans=0; for(auto p:mp){ auto v=p.second; set<int> st; for(auto x:v){ st.insert(uf2.find(x)); } ll sum=0; for(auto x:st){ sum+=uf2.size(x); } ans+=(ll)v.size()*sum; } ans-=n; cout<<ans<<endl; return 0; }