結果

問題 No.1054 Union add query
コンテスト
ユーザー ok
提出日時 2020-05-15 22:18:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 305 ms / 2,000 ms
コード長 1,367 bytes
コンパイル時間 971 ms
コンパイル使用メモリ 89,072 KB
実行使用メモリ 44,288 KB
最終ジャッジ日時 2024-09-19 11:03:34
合計ジャッジ時間 3,718 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include<iostream>
#include<string>
#include<iomanip>
#include<cmath>
#include<vector>
#include<algorithm>
#include<utility>

using namespace std;

#define int long long
#define endl "\n"

constexpr long long INF = (long long)1e18;
constexpr long long MOD = 1'000'000'007; 

struct fast_io {
	fast_io(){
		std::cin.tie(nullptr);
		std::ios::sync_with_stdio(false);
	};
} fio;


vector<int> i2g;
vector<vector<int>> g2i;
vector<int> gnum;
vector<int> num;

void init(int n) {
	i2g.resize(n);
	g2i.resize(n);
	for (int i = 0; i < n; ++i) {
		i2g[i] = i;
		g2i[i].push_back(i);
	}
}

void marge(int ia, int ib) {
	if(i2g[ia] == i2g[ib])return;
	
	if (g2i[i2g[ia]].size() < g2i[i2g[ib]].size()) {
		swap(ia, ib);
	}
	
	int ga = i2g[ia], gb = i2g[ib];
	for (int j : g2i[gb]){
		i2g[j] = ga;
		num[j] += gnum[gb] - gnum[ga];
	}
	g2i[ga].insert(g2i[ga].end(), g2i[gb].begin(), g2i[gb].end());
	g2i[gb].clear();
}

bool same(int ia, int ib) {
	return i2g[ia] == i2g[ib];
}


signed main(){
	cout<<fixed<<setprecision(10);
	
	int N, Q;
	
	cin>>N>>Q;
	
	num.resize(N);
	gnum.resize(N);
	i2g.resize(N);
	g2i.resize(N);
	
	init(N);
	
	for(int i = 0; i < Q; i++){
		int T, A, B;
		
		cin>>T>>A>>B;
		
		A--;
		
		if(T == 1) {
			B--;
			marge(A, B);
		} else if(T == 2) {
			gnum[i2g[A]] += B;
		} else if(T == 3) {
			cout<<gnum[i2g[A]] + num[A]<<endl;
		}
		
	}
	
	
	return 0;
}
0