結果
問題 | No.1054 Union add query |
ユーザー | 00 Sakuda |
提出日時 | 2024-03-31 08:45:53 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 792 ms / 2,000 ms |
コード長 | 924 bytes |
コンパイル時間 | 2,099 ms |
コンパイル使用メモリ | 207,496 KB |
実行使用メモリ | 45,000 KB |
最終ジャッジ日時 | 2024-09-30 17:51:39 |
合計ジャッジ時間 | 7,967 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 623 ms
12,672 KB |
testcase_04 | AC | 792 ms
45,000 KB |
testcase_05 | AC | 565 ms
8,192 KB |
testcase_06 | AC | 573 ms
20,844 KB |
testcase_07 | AC | 539 ms
20,840 KB |
testcase_08 | AC | 471 ms
20,844 KB |
testcase_09 | AC | 792 ms
44,416 KB |
testcase_10 | AC | 310 ms
44,356 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/dsu> using namespace atcoder; using namespace std; using ll = long long; int main() { int N, Q;cin >> N >> Q; vector<int> R(N); for (int v = 0;v < N;v++) R[v] = v; vector<vector<int>> comp(N); for (int v = 0;v < N;v++) comp[v].push_back(v); vector<ll> T(N, 0); vector<ll> S(N, 0); vector<ll> P(N, 0); while (Q--) { int t, a, b;cin >> t >> a >> b; if (t == 1) { a--;b--; if (R[a] == R[b]) continue; int ra = R[a];int rb = R[b]; if (comp[ra].size() < comp[rb].size()) swap(ra, rb); ll ad = T[rb];ll sa = T[ra]; while (!comp[rb].empty()) { int v = comp[rb].back();comp[rb].pop_back(); P[v] += ad - S[v]; S[v] = sa; R[v] = ra; comp[ra].push_back(v); } T[rb] = 0; } else if (t == 2) { a--; int r = R[a]; T[r] += b; } else { a--; int r = R[a]; ll ans = P[a] + T[r] - S[a]; printf("%lld\n", ans); } } }