結果

問題 No.1054 Union add query
ユーザー jp_stejp_ste
提出日時 2020-05-17 04:14:21
言語 Java21
(openjdk 21)
結果
AC  
実行時間 1,798 ms / 2,000 ms
コード長 1,315 bytes
コンパイル時間 2,103 ms
コンパイル使用メモリ 77,576 KB
実行使用メモリ 76,288 KB
最終ジャッジ日時 2024-09-24 10:04:27
合計ジャッジ時間 18,901 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 128 ms
54,068 KB
testcase_01 AC 138 ms
54,224 KB
testcase_02 AC 132 ms
54,076 KB
testcase_03 AC 1,710 ms
73,112 KB
testcase_04 AC 1,798 ms
75,956 KB
testcase_05 AC 1,757 ms
71,668 KB
testcase_06 AC 1,797 ms
73,656 KB
testcase_07 AC 1,708 ms
73,612 KB
testcase_08 AC 1,763 ms
71,732 KB
testcase_09 AC 1,709 ms
76,288 KB
testcase_10 AC 1,492 ms
75,644 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Arrays;
import java.util.Scanner;

public class Main {
	static Scanner scan = new Scanner(System.in);
	static UnionFind union;
	static int N, Q;
	
	public static void main(String[] args) {
		N = scan.nextInt();
		Q = scan.nextInt();
		union = new UnionFind(N);
		
		for(int i=0; i<Q; i++) {
			int T = scan.nextInt();
			int A = scan.nextInt();
			int B = scan.nextInt();
			if(T == 1) {
				union.unite(A-1, B-1);
			} else if(T == 2) {
				union.add(A-1, B);
			} else if(T == 3) {
				union.out(A-1);
			}
		}
		union.flush();
	}
}

class UnionFind {
	int[] L;
	long[] V;
	StringBuilder sb = new StringBuilder();
	
	UnionFind(int N) {
		L = new int[N];
		V = new long[N];
		Arrays.fill(L, -1);
	}
	
	int root(int x) {
		return L[x] < 0 ? x : root(L[x]);
	}

	boolean same(int x, int y) {
		return root(x) == root(y);
	}

	void unite(int x, int y) {
		x = root(x);
		y = root(y);
		if (x == y) {
			return;
		}
		if (L[x] < L[y]) {
			int w = x;
			x = y;
			y = w;
		}
		L[y] += L[x];
		L[x] = y;
		V[x] -= V[y];
	}
	
	void add(int x, int value) {
		x = root(x);
		V[x] += value;
	}
	
	void out(int x) {
		long ans = 0;
		while(true){
			ans += V[x];
			x = L[x];
			if(x < 0) break;
		}
		sb.append(ans);
		sb.append("\n");
	}
	
	void flush() {
		System.out.println(sb.toString().trim());
	}
}
0