結果
問題 | No.1054 Union add query |
ユーザー | 37zigen |
提出日時 | 2019-11-09 02:22:52 |
言語 | Java21 (openjdk 21) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,498 bytes |
コンパイル時間 | 2,685 ms |
コンパイル使用メモリ | 78,064 KB |
実行使用メモリ | 64,540 KB |
最終ジャッジ日時 | 2024-05-05 22:32:12 |
合計ジャッジ時間 | 21,187 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 146 ms
41,492 KB |
testcase_01 | AC | 149 ms
41,568 KB |
testcase_02 | AC | 135 ms
41,360 KB |
testcase_03 | TLE | - |
testcase_04 | TLE | - |
testcase_05 | TLE | - |
testcase_06 | TLE | - |
testcase_07 | TLE | - |
testcase_08 | TLE | - |
testcase_09 | TLE | - |
testcase_10 | AC | 1,558 ms
64,000 KB |
ソースコード
import java.io.*; import java.util.*; import java.math.*; public class Main { class DJSet { int n; int[] upper; long[] weight; public DJSet(int n) { this.n = n; upper = new int[n]; weight = new long[n]; Arrays.fill(upper, -1); } void add_weight(int x, long w) { x = root(x); weight[x] += w; } long get_weight(int x) { return weight[x] + (upper[x] < 0 ? 0 : get_weight(upper[x])); } int root(int x) { return upper[x] < 0 ? x : root(upper[x]); } boolean equiv(int x, int y) { return root(x) == root(y); } void setUnion(int x, int y) { x = root(x); y = root(y); if (x == y) return; if (upper[x] < upper[y]) { x ^= y; y ^= x; x ^= y; } upper[y] += upper[x]; upper[x] = y; weight[x] -= weight[y]; } } void run() { Scanner sc = new Scanner(System.in); int N = sc.nextInt(); int Q = sc.nextInt(); DJSet ds = new DJSet(N); PrintWriter pw = new PrintWriter(System.out); for (int q = 0; q < Q; ++q) { int T = sc.nextInt(); int A = sc.nextInt(); int B = sc.nextInt(); --A; if (T == 1) { --B; if (ds.equiv(A, B)) continue; ds.setUnion(A, B); } else if (T == 2) { ds.add_weight(A, B); } else if (T == 3) { pw.println(ds.get_weight(A)); } } pw.close(); } public static void main(String[] args) throws FileNotFoundException { new Main().run(); } void tr(Object... objects) { System.out.println(Arrays.deepToString(objects)); } }