結果
問題 | No.1293 2種類の道路 |
ユーザー |
![]() |
提出日時 | 2020-11-20 23:42:46 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 64 ms / 2,000 ms |
コード長 | 1,854 bytes |
コンパイル時間 | 2,504 ms |
コンパイル使用メモリ | 208,684 KB |
最終ジャッジ日時 | 2025-01-16 03:26:40 |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
#include <bits/stdc++.h>using namespace std;template <typename T, typename U> bool cmin(T &a, U b) { return a > b && (a = b, true); }template <typename T, typename U> bool cmax(T &a, U b) { return a < b && (a = b, true); }class UnionFind {private:int N;vector<int> PS;public:UnionFind() : N(0) {}UnionFind(int n) : N(n), PS(n, -1) {}bool unite(int a, int b) {int x = root(a), y = root(b);if (x == y) return false;if (-PS.at(x) < -PS.at(y)) swap(x, y);PS.at(x) += PS.at(y);PS.at(y) = x;return true;}bool same(int a, int b) { return root(a) == root(b); }int root(int a) {if (PS.at(a) < 0) return a;return PS.at(a) = root(PS.at(a));}int size(int a) { return -PS.at(root(a)); }vector<vector<int>> groups() {vector<int> B(N), S(N);for (int i = 0; i < N; i++) {B.at(i) = root(i);S.at(B.at(i))++;}vector<vector<int>> ret(N);for (int i = 0; i < N; i++) {ret.at(i).reserve(S.at(i));}for (int i = 0; i < N; i++) {ret.at(B.at(i)).push_back(i);}auto f = [&](const vector<int> &v) { return v.empty(); };ret.erase(remove_if(ret.begin(), ret.end(), f), ret.end());return ret;}};signed main() {cin.tie(nullptr);ios_base::sync_with_stdio(false);int N, D, W;cin >> N >> D >> W;UnionFind UF(N);UnionFind UF2(N);for (int i = 0; i < D; i++) {int a, b;cin >> a >> b, a--, b--;UF.unite(a, b);}for (int i = 0; i < W; i++) {int a, b;cin >> a >> b, a--, b--;UF2.unite(a, b);}long ans = 0;set<pair<int, int>> tmp;for (int i = 0; i < N; i++) {if (tmp.count({UF.root(i), UF2.root(i)})) continue;tmp.insert({UF.root(i), UF2.root(i)});ans += (long)UF.size(UF.root(i)) * UF2.size(UF2.root(i));}cout << ans - N << "\n";}