結果

問題 No.1054 Union add query
ユーザー yurujirouyurujirou
提出日時 2021-09-23 01:05:22
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,284 bytes
コンパイル時間 2,517 ms
コンパイル使用メモリ 81,860 KB
実行使用メモリ 47,072 KB
最終ジャッジ日時 2023-09-18 19:45:42
合計ジャッジ時間 8,301 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 8 ms
15,508 KB
testcase_01 AC 11 ms
15,220 KB
testcase_02 AC 11 ms
15,140 KB
testcase_03 AC 427 ms
26,968 KB
testcase_04 WA -
testcase_05 AC 403 ms
24,044 KB
testcase_06 AC 394 ms
34,160 KB
testcase_07 AC 364 ms
31,300 KB
testcase_08 AC 282 ms
31,136 KB
testcase_09 WA -
testcase_10 AC 191 ms
44,312 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <string.h>
#include <iostream>
#include <sstream>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <deque>
#include <functional>
#define _USE_MATH_DEFINES
#include <math.h>
#include <complex>
using namespace std;

//#include <atcoder/all>
//using namespace atcoder;

#define REP(i,n) for(int i = 0; i < (int)n; i++)
#define RREP(i,n) for(int i = (int)n-1; i >= 0; i--)
#define LREP(i,n) for(LL i = 0; i < (LL)n; i++)

#define Vi vector<int>
#define Vl vector<long long>
#define LP pair<long long ,long long>
#define P pair<int, int>
#define T3 tuple<LL, LL, LL>
#define T4 tuple<LL, LL, LL, LL>
#define INF 1000000007
#define SIZE	500010
#define MOD 1000000007
typedef long long LL;

int N, Q;
int T[SIZE], A[SIZE], B[SIZE], M[SIZE];
Vi E[SIZE], C;

int parent[SIZE], s[SIZE];
void init() {
	REP(i, N + 1) {
		parent[i] = i;
		s[i] = 1;
	}
}

int par(int a) {
	if (parent[a] == a) return a;
	else return parent[a] = par(parent[a]);
}

int size(int a) {
	int pa = par(a);
	return s[pa];
}

int merge(int a, int b) {
	int pa = par(a), pb = par(b);
	if (pa != pb) {
		if (size(pa) > size(pb)) swap(pa, pb);
		parent[pa] = pb;
		s[pb] += s[pa];
	}
	return pb;
}

bool same(int a, int b) {
	int pa = par(a), pb = par(b);
	if (pa == pb) return true;
	else return false;
}

int BIT[SIZE];
void add(int i, int a) {
	BIT[i] += a;
	if (i < N) add(i + (i & -i), a);
}

int sum(int i) {
	if (i == 0) return 0;
	return BIT[i] + sum(i - (i & -i));
}

int main() {
	scanf("%d %d", &N, &Q);
	REP(i, Q) {
		scanf("%d %d %d", &T[i], &A[i], &B[i]);
	}

	init();
	REP(i, N) E[i + 1].push_back(i + 1);
	REP(i, Q) {
		if (T[i] == 1) {
			int a = A[i], b = B[i];
			if (!same(a, b)) {
				int pa = par(a), pb = par(b);
				int p = merge(a, b);
				if (p != pa) swap(pa, pb);
				for (auto e : E[pb]) E[pa].push_back(e);
				E[pb].clear();
			}
		}
	}
	REP(i, N) for (auto e : E[i + 1]) C.push_back(e);
	REP(i, N) M[C[i]] = i + 1;
	init();

	REP(i, Q) {
		int a = M[A[i]], b = M[B[i]];
		if (T[i] == 1) {
			merge(a, b);
		}
		else if (T[i] == 2) {
			int left = par(a), right = left + size(a);
			add(left, B[i]);
			add(right, -B[i]);
		}
		else {
			cout << sum(a) << endl;
		}
	}
}
0