結果

問題 No.1054 Union add query
ユーザー kriiikriii
提出日時 2020-05-15 21:48:32
言語 C++17
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 370 ms / 2,000 ms
コード長 1,355 bytes
コンパイル時間 646 ms
コンパイル使用メモリ 58,824 KB
実行使用メモリ 46,836 KB
最終ジャッジ日時 2023-10-19 13:15:48
合計ジャッジ時間 3,608 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
30,012 KB
testcase_01 AC 9 ms
30,016 KB
testcase_02 AC 9 ms
30,012 KB
testcase_03 AC 205 ms
34,072 KB
testcase_04 AC 370 ms
46,836 KB
testcase_05 AC 181 ms
32,080 KB
testcase_06 AC 180 ms
37,980 KB
testcase_07 AC 168 ms
39,260 KB
testcase_08 AC 172 ms
38,368 KB
testcase_09 AC 297 ms
46,308 KB
testcase_10 AC 186 ms
45,672 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <algorithm>
#include <vector>
using namespace std;

vector<int> G[500500];
int N, Q, T[500500], A[500500], B[500500], P[500500], R[500500], S[500500], E[500500], W[500500];

int find(int x)
{
	if (P[x] != x) P[x] = find(P[x]);
	return P[x];
}

void uni(int x, int y)
{
	x = find(x);
	y = find(y);
	if (x == y) return;
	P[x] = y;
	S[y] = min(S[y], S[x]);
	E[y] = max(E[y], E[x]);
}

void push(int x, int p)
{
	while (x <= N){
		W[x] += p;
		x += x & (-x);
	}
}

int pop(int x)
{
	int r = 0;
	while (x){
		r += W[x];
		x -= x & (-x);
	}
	return r;
}

int main()
{
	scanf("%d %d", &N, &Q);
	for (int i = 0; i < Q; i++) scanf ("%d %d %d", &T[i], &A[i], &B[i]);
	for (int i = 1; i <= N; i++) P[i] = i, G[i] = { i };
	for (int i = 0; i < Q; i++) if (T[i] == 1){
		int x = P[A[i]], y = P[B[i]];
		if (x == y) continue;
		if (G[x].size() > G[y].size()) swap(x, y);
		G[y].insert(G[y].end(), G[x].begin(), G[x].end());
		for (auto &v : G[x]) P[v] = y;
		G[x].clear();
	}
	for (int i = 1, c = 1; i <= N; i++) for (auto &v : G[i]) R[v] = c++;
	for (int i = 1; i <= N; i++){
		P[i] = i; S[i] = R[i]; E[i] = R[i] + 1;
	}
	for (int i = 0; i < Q; i++){
		if (T[i] == 1) uni(A[i], B[i]);
		else if (T[i] == 2){
			int p = find(A[i]);
			push(S[p], B[i]);
			push(E[p], -B[i]);
		}
		else{
			printf ("%d\n", pop(R[A[i]]));
		}
	}

	return 0;
}
0