結果

問題 No.1054 Union add query
ユーザー yudedako
提出日時 2020-08-27 14:28:21
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 175 ms / 2,000 ms
コード長 1,589 bytes
コンパイル時間 1,000 ms
コンパイル使用メモリ 110,756 KB
実行使用メモリ 7,040 KB
最終ジャッジ日時 2024-11-07 17:29:50
合計ジャッジ時間 4,534 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <utility>
#include <tuple>
#include <vector>
#include <string>
#include <map>
#include <unordered_map>
#include <set>
#include <unordered_set>
#include <algorithm>
#include <functional>
#include <climits>
#include <numeric>
#include <queue>
#include <cmath>
#include <iomanip>
#include <array>
#include <string>
#include <stack>
#include <cassert>
#include <memory>
#include <random>
#include <fstream>
#include <cfloat>
#include <complex>



class UnionFind {
	std::vector<int> parent, added;
public:
	UnionFind(int size): parent(size, -1), added(size, 0){}
	int find(int a) {
		if (parent[a] < 0) {
			return a;
		}
		else {
			const auto root = find(parent[a]);
			if (parent[a] != root) {
				added[a] += added[parent[a]];
				parent[a] = root;
			}
			return root;
		}
	}
	void add(int node, int value) {
		added[find(node)] += value;
	}
	void unite(int a, int b) {
		a = find(a);
		b = find(b);
		if (a == b) return;
		if (parent[a] > parent[b]) std::swap(a, b);
		parent[a] += parent[b];
		parent[b] = a;
		added[b] -= added[a];
	}
	int value_of(int a) {
		if (parent[a] < 0) {
			return added[a];
		}
		else {
			return added[find(a)] + added[a];
		}
	}
};

int main() {
	std::cin.tie(nullptr);
	std::ios_base::sync_with_stdio(false);
	int n, q; std::cin >> n >> q;
	UnionFind uft(n);
	for (auto i = 0; i < q; ++i) {
		int t, a, b; std::cin >> t >> a >> b;
		switch(t) {
			case 1:
				uft.unite(a - 1, b - 1); break;
			case 2:
				uft.add(a - 1, b); break;
			case 3:
				std::cout << uft.value_of(a - 1) << '\n'; break;
			default: throw 0;
		}
	}
}
0