結果
問題 | No.1054 Union add query |
ユーザー | GGanari |
提出日時 | 2024-03-20 11:16:18 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 201 ms / 2,000 ms |
コード長 | 1,822 bytes |
コンパイル時間 | 3,185 ms |
コンパイル使用メモリ | 227,104 KB |
実行使用メモリ | 11,136 KB |
最終ジャッジ日時 | 2024-09-30 06:02:33 |
合計ジャッジ時間 | 6,744 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
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 | 161 ms
5,248 KB |
testcase_04 | AC | 201 ms
11,008 KB |
testcase_05 | AC | 148 ms
5,248 KB |
testcase_06 | AC | 132 ms
6,400 KB |
testcase_07 | AC | 121 ms
6,528 KB |
testcase_08 | AC | 133 ms
6,400 KB |
testcase_09 | AC | 172 ms
11,136 KB |
testcase_10 | AC | 101 ms
11,008 KB |
ソースコード
#pragma GCC optimize("O3") #pragma GCC optimize("Ofast") #pragma GCC optimize("unroll-loops") #pragma GCC target("avx,avx2") #include <bits/stdc++.h> #define INF 1000000001LL #define LNF 1000000000000000001LL #define MOD 1000000007LL #define MAX 502 #define long long long #define all(x) x.begin(),x.end() using namespace std; //UnionFind class UnionFind { public: vector<int> parent; vector<int> cnt; vector<long> value; void init(int size) { parent.resize(size); cnt.resize(size); value.resize(size); for(int i = 0; i<size; i++) { parent[i] = i; cnt[i] = 1; } } int find(int x) { if(parent[x] == x) return x; return find(parent[x]); } void merge(int x, int y) { x = find(x); y = find(y); if(x == y) return; if(cnt[x] == cnt[y]) cnt[x]++; if(cnt[x] > cnt[y]) { parent[y] = x; value[y]-=value[x]; } else { parent[x] = y; value[x]-=value[y]; } } void update(int x, int v) { x = find(x); value[x]+=v; } long findSum(int x) { if(parent[x] == x) return value[x]; return findSum(parent[x])+value[x]; } }; int main() { ios_base::sync_with_stdio(0); cin.tie(0); int n,q; cin >> n >> q; UnionFind uf; uf.init(n+1); while(q--) { int t,a,b; cin >> t >> a >> b; if(t == 1) { uf.merge(a,b); } if(t == 2) { uf.update(a,b); } if(t == 3) { cout << uf.findSum(a) << "\n"; } } return 0; }