結果
問題 | No.2290 UnUnion Find |
ユーザー | とりゐ |
提出日時 | 2023-05-05 21:30:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 671 ms / 2,000 ms |
コード長 | 1,943 bytes |
コンパイル時間 | 626 ms |
コンパイル使用メモリ | 87,132 KB |
実行使用メモリ | 129,876 KB |
最終ジャッジ日時 | 2023-08-15 03:03:40 |
合計ジャッジ時間 | 26,506 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 95 ms
71,800 KB |
testcase_01 | AC | 92 ms
71,784 KB |
testcase_02 | AC | 252 ms
102,000 KB |
testcase_03 | AC | 300 ms
107,320 KB |
testcase_04 | AC | 299 ms
107,452 KB |
testcase_05 | AC | 302 ms
107,072 KB |
testcase_06 | AC | 298 ms
107,408 KB |
testcase_07 | AC | 294 ms
107,412 KB |
testcase_08 | AC | 294 ms
107,280 KB |
testcase_09 | AC | 291 ms
107,316 KB |
testcase_10 | AC | 290 ms
107,448 KB |
testcase_11 | AC | 293 ms
107,420 KB |
testcase_12 | AC | 299 ms
107,496 KB |
testcase_13 | AC | 298 ms
107,508 KB |
testcase_14 | AC | 303 ms
107,440 KB |
testcase_15 | AC | 308 ms
107,304 KB |
testcase_16 | AC | 302 ms
107,312 KB |
testcase_17 | AC | 298 ms
106,936 KB |
testcase_18 | AC | 294 ms
107,480 KB |
testcase_19 | AC | 612 ms
119,136 KB |
testcase_20 | AC | 577 ms
126,228 KB |
testcase_21 | AC | 417 ms
104,512 KB |
testcase_22 | AC | 521 ms
108,936 KB |
testcase_23 | AC | 509 ms
108,120 KB |
testcase_24 | AC | 598 ms
125,764 KB |
testcase_25 | AC | 514 ms
108,372 KB |
testcase_26 | AC | 578 ms
129,876 KB |
testcase_27 | AC | 671 ms
126,616 KB |
testcase_28 | AC | 589 ms
112,672 KB |
testcase_29 | AC | 646 ms
122,540 KB |
testcase_30 | AC | 483 ms
107,188 KB |
testcase_31 | AC | 633 ms
119,556 KB |
testcase_32 | AC | 519 ms
107,940 KB |
testcase_33 | AC | 561 ms
112,212 KB |
testcase_34 | AC | 565 ms
129,436 KB |
testcase_35 | AC | 650 ms
129,160 KB |
testcase_36 | AC | 500 ms
108,184 KB |
testcase_37 | AC | 589 ms
111,584 KB |
testcase_38 | AC | 537 ms
108,244 KB |
testcase_39 | AC | 531 ms
109,268 KB |
testcase_40 | AC | 649 ms
127,492 KB |
testcase_41 | AC | 528 ms
107,768 KB |
testcase_42 | AC | 617 ms
116,432 KB |
testcase_43 | AC | 643 ms
116,048 KB |
testcase_44 | AC | 309 ms
107,464 KB |
testcase_45 | AC | 296 ms
110,812 KB |
testcase_46 | AC | 305 ms
110,436 KB |
ソースコード
from sys import stdin input=lambda :stdin.readline()[:-1] from collections import defaultdict class UnionFind(): def __init__(self,n): self.n=n self.parents=[-1]*n self.grp_cnt=n def find(self,x): if self.parents[x]<0: return x else: self.parents[x]=self.find(self.parents[x]) return self.parents[x] def union(self,x,y): x=self.find(x) y=self.find(y) if x==y: return if self.parents[x]>self.parents[y]: x,y=y,x self.parents[x]+=self.parents[y] self.parents[y]=x self.grp_cnt-=1 def size(self,x): return -self.parents[self.find(x)] def same(self,x,y): return self.find(x)==self.find(y) def members(self,x): root=self.find(x) return [i for i in range(self.n) if self.find(i)==root] def roots(self): return [i for i, x in enumerate(self.parents) if x< 0] def group_count(self): return len(self.roots()) def all_group_members(self): group_members=defaultdict(list) for member in range(self.n): group_members[self.find(member)].append(member) return group_members n,q=map(int,input().split()) if n==1: for i in range(q): Q=list(map(int,input().split())) if Q[0]==2: print(-1) exit() query=[] u0,v0=-1,-1 uf=UnionFind(n) for _ in range(q): Q=list(map(int,input().split())) if Q[0]==1: u,v=Q[1:] u-=1 v-=1 uf.union(u,v) if u0==-1 and uf.grp_cnt==2: g=uf.all_group_members() for i in g: if u0==-1: u0=g[i][0] else: v0=g[i][0] query.append((1,u,v)) else: query.append((2,Q[1]-1,-1)) if u0==-1: g=uf.all_group_members() for i in g: if u0==-1: u0=g[i][0] elif v0==-1: v0=g[i][0] uf=UnionFind(n) for t,u,v in query: if t==1: uf.union(u,v) else: if not uf.same(u,u0): print(u0+1) elif not uf.same(u,v0): print(v0+1) else: print(-1)