結果

問題 No.1293 2種類の道路
ユーザー ninoinui
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#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";
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0