結果
問題 | No.1054 Union add query |
ユーザー | 37zigen |
提出日時 | 2019-11-09 04:23:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 635 ms / 2,000 ms |
コード長 | 1,480 bytes |
コンパイル時間 | 563 ms |
コンパイル使用メモリ | 82,232 KB |
実行使用メモリ | 86,628 KB |
最終ジャッジ日時 | 2024-05-05 22:33:51 |
合計ジャッジ時間 | 4,536 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 37 ms
51,840 KB |
testcase_01 | AC | 38 ms
51,968 KB |
testcase_02 | AC | 38 ms
51,712 KB |
testcase_03 | AC | 556 ms
81,560 KB |
testcase_04 | AC | 403 ms
86,628 KB |
testcase_05 | AC | 635 ms
81,968 KB |
testcase_06 | AC | 213 ms
79,232 KB |
testcase_07 | AC | 177 ms
79,460 KB |
testcase_08 | AC | 201 ms
79,252 KB |
testcase_09 | AC | 247 ms
84,480 KB |
testcase_10 | AC | 127 ms
82,688 KB |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines class DJSet(): def __init__(self, n): self.upper = [-1]*n self.weight = [0]*n def equiv(self, x, y): return self.root(x) == self.root(y) def root(self, x): if self.upper[x] < 0: return x else: return self.root(self.upper[x]) def get_weight(self, x): if self.upper[x] < 0: return self.weight[x] else: return self.weight[x] + self.get_weight(self.upper[x]) def setUnion(self, x, y): x = self.root(x) y = self.root(y) if x == y: return if self.upper[x] < self.upper[y]: x ^= y y ^= x x ^= y self.upper[y] = self.upper[y] + self.upper[x] self.upper[x] = y self.weight[x] = self.weight[x] - self.weight[y] def add_weight(self, x, w): x = self.root(x) self.weight[x] = self.weight[x] + w def run(): N, Q = map(int, readline().split()) ds = DJSet(N) for q in range(Q): T, A, B = map(int, readline().split()) A = A - 1 if (T == 1): B = B - 1 if (ds.equiv(A, B)): continue ds.setUnion(A, B) elif (T == 2): ds.add_weight(A, B) elif (T == 3): print(ds.get_weight(A)) run()