結果

問題 No.1054 Union add query
ユーザー kuhakukuhaku
提出日時 2020-05-17 11:18:10
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 166 ms / 2,000 ms
コード長 2,919 bytes
コンパイル時間 2,068 ms
コンパイル使用メモリ 207,460 KB
実行使用メモリ 11,108 KB
最終ジャッジ日時 2024-09-25 02:08:22
合計ジャッジ時間 4,501 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 146 ms
6,940 KB
testcase_04 AC 166 ms
11,108 KB
testcase_05 AC 134 ms
6,944 KB
testcase_06 AC 131 ms
6,940 KB
testcase_07 AC 113 ms
6,944 KB
testcase_08 AC 130 ms
6,940 KB
testcase_09 AC 164 ms
10,992 KB
testcase_10 AC 93 ms
11,012 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
confirm 0LL and 1LL
confirm cornercases such as 0
confirm times of cin < 10^6
*/
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
using P = pair<ll, ll>;
using Pld = pair<ld, ld>;
using Vec = vector<ll>;
using VecP = vector<P>;
using VecB = vector<bool>;
using VecC = vector<char>;
using VecD = vector<ld>;
using VecS = vector<string>;
using Graph = vector<VecP>;
template <typename T>
using Vec1 = vector<T>;
template <typename T>
using Vec2 = vector<Vec1<T> >;
#define REP(i, m, n) for(ll i = (m); (i) < (n); ++(i))
#define REPN(i, m, n) for(ll i = (m); (i) <= (n); ++(i))
#define REPR(i, m, n) for(ll i = (m)-1; (i) >= (n); --(i))
#define REPNR(i, m, n) for(ll i = (m); (i) >= (n); --(i))
#define rep(i, n) REP(i, 0, n)
#define repn(i, n) REPN(i, 1, n)
#define repr(i, n) REPR(i, n, 0)
#define repnr(i, n) REPNR(i, n, 1)
#define all(s) (s).begin(), (s).end()
#define pb push_back
#define fs first
#define sc second
template <typename T>
bool chmax(T &a, const T b){if(a < b){a = b; return true;} return false;}
template <typename T>
bool chmin(T &a, const T b){if(a > b){a = b; return true;} return false;}
template <typename T>
ll pow2(const T n){return (1LL << n);}
template <typename T>
void cosp(const T n){cout << n << ' ';}
void co(void){cout << '\n';}
template <typename T>
void co(const T n){cout << n << '\n';}
template <typename T1, typename T2>
void co(pair<T1, T2> p){cout << p.fs << ' ' << p.sc << '\n';}
template <typename T>
void co(const Vec1<T> &v){for(T i : v) cosp(i); co();}
template <typename T>
void co(initializer_list<T> v){for(T i : v) cosp(i); co();}
template <typename T>
void ce(const T n){cerr << n << endl;}
void sonic(){ios::sync_with_stdio(false); cin.tie(0);}
void setp(const ll n){cout << fixed << setprecision(n);}
constexpr int INF = 1e9+1;
constexpr ll LINF = 1e18+1;
constexpr ll MOD = 1e9+7;
//constexpr ll MOD = 998244353;
constexpr ld EPS = 1e-11;
const ld PI = acos(-1);

struct UnionFind{
	Vec par;
	Vec sum;

	UnionFind(ll n){
		par.resize(n);
		sum.assign(n, 0);
		rep(i, n) par[i] = i;
	}

	ll root(ll x){
		if(par[x] == x) return x;
		ll p = root(par[x]);
		if(p != par[x]) sum[x] += sum[par[x]];
		return par[x] = p;
	}

	void unite(ll x, ll y){
		x = root(x), y = root(y);
		if(x != y){
			par[y] = x;
			sum[y] -= sum[x];
		}
	}

	void add(ll x, ll a){
		sum[root(x)] += a;
	}

	ll getSum(ll x){
		if(x == root(x)) return sum[x];
		return sum[x] + sum[par[x]];
	}

	bool same(ll x, ll y){
		return root(x) == root(y);
	}
};

int main(void){
	sonic();
	ll n, q;
	cin >> n >> q;

	UnionFind uf(n);

	rep(i, q){
		ll t, a, b;
		cin >> t >> a >> b;
		if(t == 1){
			a--, b--;
			uf.unite(a, b);
		}else if(t == 2){
			a--;
			uf.add(a, b);
		}else{
			a--;
			co(uf.getSum(a));
		}
	}

	return 0;
}
0