結果
問題 | No.1054 Union add query |
ユーザー | ntuda |
提出日時 | 2021-10-09 12:26:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 844 ms / 2,000 ms |
コード長 | 1,015 bytes |
コンパイル時間 | 626 ms |
コンパイル使用メモリ | 87,308 KB |
実行使用メモリ | 139,152 KB |
最終ジャッジ日時 | 2023-10-10 21:51:30 |
合計ジャッジ時間 | 8,806 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 72 ms
71,292 KB |
testcase_01 | AC | 72 ms
71,484 KB |
testcase_02 | AC | 72 ms
71,444 KB |
testcase_03 | AC | 844 ms
135,936 KB |
testcase_04 | AC | 707 ms
139,152 KB |
testcase_05 | AC | 778 ms
133,888 KB |
testcase_06 | AC | 563 ms
132,740 KB |
testcase_07 | AC | 541 ms
132,888 KB |
testcase_08 | AC | 556 ms
132,592 KB |
testcase_09 | AC | 589 ms
137,748 KB |
testcase_10 | AC | 524 ms
136,560 KB |
ソースコード
import sys sys.setrecursionlimit(500000) def find(x): if par[x] == x: return x else: # par[x] = find(par[x]) #経路圧縮 # return par[x] return find(par[x]) #経路圧縮なし def same(x,y): return find(x) == find(y) def unite(x,y): x = find(x) y = find(y) if x == y: return 0 if x < y: par[y] = x else: par[x] = y N,Q = map(int,input().split()) TAB = [list(map(int,input().split())) for _ in range(Q)] par = [i for i in range(N+1)] B = [0] * (N+1) for i,(t,a,b) in enumerate(TAB): if t == 1: pa = find(a) pb = find(b) if pa == pb: continue if pa < pb: B[pb] -= B[pa] else: B[pa] -= B[pb] unite(pa,pb) elif t==2: pa = find(a) B[pa] += b else: ans = 0 while 1: ans += B[a] c = par[a] if a == c: break a = c print(ans)