#483342 (C++17) No.1054 Union add query

• 問題: No.1054 Union add query
• ユーザー: define
• 提出日時: 2020-05-18 15:42:11
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 345 ms / 2,000 ms
• コード長: 1,257 bytes
• コンパイル時間: 2,391 ms
• コンパイル使用メモリ: 207,376 KB
• 実行使用メモリ: 44,520 KB
• 最終ジャッジ日時: 2024-10-01 21:56:55
• 合計ジャッジ時間: 6,505 ms
• ジャッジサーバーID （参考情報）: judge4 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	8 ms; 17,600 KB
testcase_01	AC	8 ms; 19,428 KB
testcase_02	AC	7 ms; 15,812 KB
testcase_03	AC	200 ms; 22,684 KB
testcase_04	AC	345 ms; 44,520 KB
testcase_05	AC	156 ms; 19,656 KB
testcase_06	AC	169 ms; 32,096 KB
testcase_07	AC	143 ms; 32,080 KB
testcase_08	AC	162 ms; 31,820 KB
testcase_09	AC	261 ms; 39,388 KB
testcase_10	AC	130 ms; 36,412 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define rep(i,n) for(int i=0;i<n;i++)
#define REP(i,n) for(int i=1;i<n;i++)
#define all(v) v.begin(),v.end()
#define P pair<int,int>
#define len(s) (int)s.size()
#define pb push_back

template<class T> inline bool chmin(T &a, T b){
	if(a>b){a=b;return true;}
	return false;
}
template<class T> inline bool chmax(T &a, T b){
	if(a<b){a=b;return true;}
	return false;
}
constexpr int mod = 1e9+7;
constexpr int inf = 3e18;

int N,Q;
vector<int>v[500005];
int ans[500005],add[500005],par[500005];
int find(int x){
	if(par[x]==x)return x;
	return par[x]=find(par[x]);
}
signed main(){
	cin.tie(0);ios::sync_with_stdio(false);
	cin>>N>>Q;
	rep(i,N){
		par[i]=i;v[i].pb(i);
	}
	while(Q--){
		int t,a,b;cin>>t>>a>>b;
		if(t==1){
			a--;b--;
			a=find(a);b=find(b);
			if(a==b)continue;
			if(len(v[a])<len(v[b])){
				for(int i:v[a])ans[i]+=add[a]-add[b];
				add[a]=0;
				v[b].insert(v[b].end(),all(v[a]));
				v[a].clear();
				par[a]=b;
			}else {
				for(int i:v[b])ans[i]+=add[b]-add[a];
				add[b]=0;
				v[a].insert(v[a].end(),all(v[b]));
				v[b].clear();
				par[b]=a;
			}
		}else if(t==2){
			a--;a=find(a);
			add[a]+=b;
		}else {
			a--;
			cout<<ans[a]+add[find(a)]<<"\n";
		}
	}
}