結果
問題 | No.1054 Union add query |
ユーザー |
|
提出日時 | 2020-05-15 21:44:43 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 175 ms / 2,000 ms |
コード長 | 1,428 bytes |
コンパイル時間 | 2,271 ms |
コンパイル使用メモリ | 195,580 KB |
最終ジャッジ日時 | 2025-01-10 11:27:01 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 |
コンパイルメッセージ
main.cpp: In function ‘int main()’: main.cpp:59:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result] 59 | scanf("%d %d %d", &t, &a, &b); | ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
ソースコード
#include <bits/stdc++.h> using namespace std; struct UnionFind { vector<int> par; vector<int> sz; vector<int> V; UnionFind(int n=0){ if(n>0) initialize(n); } void initialize(int n){ par.resize(n); sz.assign(n, 1); V.assign(n, 0); for(int i=0; i<n; i++){ par[i] = i; } } int find(int x){ if(par[x] == x){ return x; }else{ return find(par[x]); } } int ans(int x){ if(par[x] == x){ return V[x]; }else{ return V[x] + ans(par[x]); } } bool unite(int x, int y){ x = find(x), y = find(y); if(x == y) return false; if(sz[x] > sz[y]) swap(x, y); par[x] = y; sz[y] += sz[x]; V[x] -= V[y]; return true; } bool same(int x, int y){ return find(x) == find(y); } int size(int x){ return sz[find(x)]; } void add(int k, int v){ V[find(k)] += v; } }; int main(){ int N, Q; cin >> N >> Q; UnionFind uf(N); while(Q--){ int t, a, b; scanf("%d %d %d", &t, &a, &b); if(t == 1){ a--; b--; uf.unite(a, b); }else if(t == 2){ a--; uf.add(a, b); }else{ a--; int ans = uf.ans(a); printf("%d\n", ans); } } return 0; }