結果
問題 | No.1293 2種類の道路 |
ユーザー |
|
提出日時 | 2020-12-08 21:05:24 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 60 ms / 2,000 ms |
コード長 | 2,548 bytes |
コンパイル時間 | 1,329 ms |
コンパイル使用メモリ | 138,144 KB |
最終ジャッジ日時 | 2025-01-16 19:56:03 |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <cmath>#include <queue>#include <string>#include <map>#include <set>#include <stack>#include <tuple>#include <deque>#include <array>#include <numeric>#include <bitset>#include <iomanip>#include <cassert>#include <chrono>#include <random>#include <limits>#include <iterator>#include <functional>#include <sstream>#include <fstream>#include <complex>#include <cstring>#include <unordered_map>using namespace std;using ll = long long;using P = pair<int, int>;constexpr int INF = 1001001001;constexpr int mod = 1000000007;// constexpr int mod = 998244353;template<class T>inline bool chmax(T& x, T y){if(x < y){x = y;return true;}return false;}template<class T>inline bool chmin(T& x, T y){if(x > y){x = y;return true;}return false;}struct CountUnionFind{int sz; // vertex numbervector<int> par;vector<int> rank;vector<int> cnt;CountUnionFind(int n) : sz(n) {par.resize(sz);rank.assign(sz, 0);cnt.assign(sz, 1);for(int i = 0; i < sz; ++i){par[i] = i;}}int find(int x){if(par[x] == x) return x;else return par[x] = find(par[x]);}bool same(int x, int y){return find(x) == find(y);}void unite(int x, int y){x = find(x), y = find(y);if(x == y) return;if(rank[x] < rank[y]){par[x] = y;cnt[y] += cnt[x];}else{par[y] = x;cnt[x] += cnt[y];if(rank[x] == rank[y]) ++rank[x];}}int get_cnt(int x){return cnt[find(x)];}};int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int N, D, W;cin >> N >> D >> W;CountUnionFind cuf(N), wuf(N);for(int i = 0; i < D; ++i){int u, v;cin >> u >> v;--u, --v;cuf.unite(u, v);}for(int i = 0; i < W; ++i){int u, v;cin >> u >> v;--u, --v;wuf.unite(u, v);}vector<set<int>> dat(N);for(int i = 0; i < N; ++i){dat[cuf.find(i)].emplace(wuf.find(i));}vector<int> sum(N);for(int i = 0; i < N; ++i){for(int par : dat[i]) sum[i] += wuf.get_cnt(par);}ll ans = 0;for(int i = 0; i < N; ++i){if(sum[i] > 0) ans += (ll)cuf.get_cnt(i) * (sum[i] - 1);}cout << ans << endl;}