結果
問題 |
No.3200 Sinking Islands
|
ユーザー |
![]() |
提出日時 | 2025-07-24 22:53:48 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 111 ms / 2,000 ms |
コード長 | 1,171 bytes |
コンパイル時間 | 2,173 ms |
コンパイル使用メモリ | 201,384 KB |
実行使用メモリ | 14,208 KB |
最終ジャッジ日時 | 2025-07-24 22:53:57 |
合計ジャッジ時間 | 8,841 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
#include <bits/stdc++.h> using namespace std; #define int ll #define fast_io cin.tie(0)->sync_with_stdio(0); #define endl '\n' typedef long long ll; int32_t main() { fast_io; int n, m; cin >> n >> m; vector<pair<int, int>> edges(m); for (auto &[a, b] : edges) cin >> a >> b, --a, --b; int q; cin >> q; vector<int> queries(q); for (auto &i : queries) cin >> i, --i; vector<int> when(m, -1); for (int i = 0; i < q; i++) when[queries[i]] = i; vector<int> id(n), sz(n, 1); iota(id.begin(), id.end(), 0); function<int(int)> vai = [&] (int i) { if (i == id[i]) return i; return id[i] = vai(id[i]); }; auto unite = [&] (int a, int b) -> int { a = vai(a), b = vai(b); if (a == b) return 0; int qt = sz[a] * sz[b]; sz[a] += sz[b]; id[b] = a; return qt; }; int tot = n * (n - 1) / 2; for (int i = 0; i < m; i++) if (when[i] == -1) { auto [a, b] = edges[i]; int qt = unite(a, b); tot -= qt; } vector<int> ans(q); for (int i = q - 1; i >= 0; --i) { auto [a, b] = edges[queries[i]]; ans[i] = unite(a, b); tot -= ans[i]; } for (int i = 0; i < q; i++) { if (i) ans[i] += ans[i - 1]; cout << ans[i] + tot << endl; } }