結果

問題 No.3200 Sinking Islands
ユーザー iastm
提出日時 2025-07-12 03:11:57
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 491 ms / 2,000 ms
コード長 1,302 bytes
コンパイル時間 1,389 ms
コンパイル使用メモリ 120,524 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2025-07-12 03:12:10
合計ジャッジ時間 11,616 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <vector>
#include <stack>
#include <atcoder/dsu>

int main() {
    int N, M;
    std::cin >> N >> M;
    std::vector<int> U(M), V(M);
    std::vector<bool> usable(M, true);
    for (int i = 0; i < M; i++) {
        int u, v;
        std::cin >> u >> v;
        u--;
        v--;
        U[i] = u;
        V[i] = v;
    }
    int Q;
    std::cin >> Q;
    std::stack<int> queries;
    while (Q--) {
        int B;
        std::cin >> B;
        B--;
        queries.push(B);
        usable[B] = false;
    }
    atcoder::dsu dsu(N);
    for (int i = 0; i < M; i++) {
        if (usable[i]) dsu.merge(U[i], V[i]);
    }
    int64_t count = 0;
    std::stack<int64_t> output;
    for (int i = 0; i < N; i++) {
        if (dsu.leader(i) == i) count += dsu.size(i) * int64_t(dsu.size(i) - 1) / 2;
    }
    output.push(count);
    while (queries.size() > 1) {
        int i = queries.top();
        queries.pop();
        if (dsu.same(U[i], V[i])) {
            output.push(count);
        } else {
            count += int64_t(dsu.size(U[i])) * dsu.size(V[i]);
            output.push(count);
            dsu.merge(U[i], V[i]);
        }
    }
    while (!output.empty()) {
        std::cout << N * int64_t(N - 1) / 2 - output.top() << std::endl;
        output.pop();
    }
}
0