結果

問題 No.1293 2種類の道路
ユーザー commycommy
提出日時 2020-11-28 20:36:48
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 70 ms / 2,000 ms
コード長 2,555 bytes
コンパイル時間 1,188 ms
コンパイル使用メモリ 89,232 KB
最終ジャッジ日時 2025-01-16 09:57:53
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>
#include <numeric>
#include <utility>
#include <tuple>

#define rep(i, a, b) for (int i = int(a); i < int(b); i++)
using namespace std;
using ll = long long int;
using P = pair<ll, ll>;

// clang-format off
#ifdef _DEBUG_
#define dump(...) do{ cerr << __LINE__ << ":\t" << #__VA_ARGS__ << " = "; PPPPP(__VA_ARGS__); cerr << endl; } while(false)
template<typename T> void PPPPP(T t) { cerr << t; }
template<typename T, typename... S> void PPPPP(T t, S... s) { cerr << t << ", "; PPPPP(s...); }
#else
#define dump(...) do{ } while(false)
#endif
template<typename T> vector<T> make_v(size_t a, T b) { return vector<T>(a, b); }
template<typename... Ts> auto make_v(size_t a, Ts... ts) { return vector<decltype(make_v(ts...))>(a, make_v(ts...)); }
template<typename T> bool chmin(T &a, T b) { if (a > b) {a = b; return true; } return false; }
template<typename T> bool chmax(T &a, T b) { if (a < b) {a = b; return true; } return false; }
template<typename T> void print(T a) { cout << a << '\n'; }
template<typename T, typename... Ts> void print(T a, Ts... ts) { cout << a << ' '; print(ts...); }
template<typename T> istream &operator,(istream &in, T &t) { return in >> t; }
// clang-format on

class UnionFind {
public:
    vector<int> par;
    int N;
    UnionFind(int n) {
        par.resize(n, -1);
        N = n;
    }
    int Find(int n) {
        return par[n] < 0 ? n : par[n] = Find(par[n]);
    }
    bool Union(int x, int y) {
        x = Find(x);
        y = Find(y);
        if (x == y) return false;
        if (-par[x] > -par[y]) swap(x, y);
        par[y] += par[x];
        par[x] = y;
        N--;
        return true;
    }
    bool Same(int x, int y) {
        return Find(x) == Find(y);
    }
    int size(int x) {
        return -par[Find(x)];
    }
    int size() {
        return N;
    }
};

#include <set>

int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n, d, w;
    cin, n, d, w;
    UnionFind uf1(n), uf2(n);
    rep(i, 0, d) {
        int a, b;
        cin, a, b;
        a--;
        b--;
        uf1.Union(a, b);
    }
    rep(i, 0, w) {
        int a, b;
        cin, a, b;
        a--;
        b--;
        uf2.Union(a, b);
    }
    set<P> used;
    ll ans = 0;
    rep(i, 0, n) {
        int p1 = uf1.Find(i);
        int p2 = uf2.Find(i);
        if (!used.count(P(p1, p2))) {
            ans += 1LL * uf1.size(i) * uf2.size(i);
            used.insert(P(p1, p2));
        }
    }
    print(ans - n);
    return 0;
}
0