結果
問題 | No.1054 Union add query |
ユーザー | 00 Sakuda |
提出日時 | 2024-03-31 08:45:02 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 806 ms / 2,000 ms |
コード長 | 922 bytes |
コンパイル時間 | 2,146 ms |
コンパイル使用メモリ | 209,444 KB |
実行使用メモリ | 45,076 KB |
最終ジャッジ日時 | 2024-03-31 08:45:11 |
合計ジャッジ時間 | 9,434 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,676 KB |
testcase_01 | AC | 2 ms
6,676 KB |
testcase_02 | AC | 1 ms
6,676 KB |
testcase_03 | AC | 656 ms
12,672 KB |
testcase_04 | AC | 806 ms
45,076 KB |
testcase_05 | AC | 616 ms
8,064 KB |
testcase_06 | AC | 593 ms
20,968 KB |
testcase_07 | AC | 582 ms
20,968 KB |
testcase_08 | AC | 488 ms
20,968 KB |
testcase_09 | AC | 787 ms
44,308 KB |
testcase_10 | AC | 344 ms
44,308 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]; cout << ans << endl; } } }