結果

問題 No.1293 2種類の道路
コンテスト
ユーザー rabimea
提出日時 2026-01-22 02:05:09
言語 C++23
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 124 ms / 2,000 ms
コード長 1,525 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,884 ms
コンパイル使用メモリ 350,620 KB
実行使用メモリ 15,548 KB
最終ジャッジ日時 2026-01-22 02:05:16
合計ジャッジ時間 6,790 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>

#define rep(i, a, b) for(int i = (a); i <= (b); i ++)
using std::cin, std::cout, std::cerr;
using ll = long long;

struct DSU {
    int n;
    std::vector<int> p, size;
    DSU(int n) { Init(n); }
    void Init(int n) {
        this->n = n;
        p.resize(n);
        size.assign(n, 1);
        std::iota(p.begin(), p.end(), 0);
    }
    int operator ()(int x) {
        if(x == p[x]) return x;
        return p[x] = this->operator()(p[x]);
    }
    void Link(int x, int y) {
        size[y] += size[x];
        p[x] = y;
    }
};

int main() {
    std::ios::sync_with_stdio(false);
    int n, m, k; cin >> n >> m >> k;
    std::vector<std::vector<int>> e(n + 1);
    rep(i, 1, m) {
        int x, y; cin >> x >> y;
        e[x].push_back(y);
        e[y].push_back(x);
    }
    DSU dsu(n + 1);
    rep(i, 1, k) {
        int x, y; cin >> x >> y;
        x = dsu(x);
        y = dsu(y);
        if(x != y)
            dsu.Link(x, y);
    }

    std::vector<int> v;
    std::vector<bool> vis(n + 1);
    std::function<void(int)> dfs = [&](int x) {
        vis[x] = 1;
        v.push_back(x);
        for(int i : e[x]) if(!vis[i])
            dfs(i);
    };

    ll ans = 0;
    rep(i, 1, n) if(!vis[i]) {
        v.clear();
        dfs(i);

        std::set<int> set;
        for(int x : v) {
            int y = dsu(x);
            if(!set.count(y)) {
                set.insert(y);
                ans += ll(v.size()) * dsu.size[y];
            }
        }
    }
    cout << ans - n << '\n';
}
0