結果
問題 | No.2290 UnUnion Find |
ユーザー |
|
提出日時 | 2023-05-05 21:47:26 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 851 ms / 2,000 ms |
コード長 | 1,161 bytes |
コンパイル時間 | 412 ms |
コンパイル使用メモリ | 82,540 KB |
実行使用メモリ | 110,464 KB |
最終ジャッジ日時 | 2024-11-23 06:49:29 |
合計ジャッジ時間 | 33,401 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 46 |
ソースコード
class Unionfind:def __init__(self,n):self.uf = [-1]*nself.ps = set([i for i in range(n)])def find(self,x):if self.uf[x] < 0:return xelse:self.uf[x] = self.find(self.uf[x])return self.uf[x]def same(self,x,y):return self.find(x) == self.find(y)def union(self,x,y):x = self.find(x)y = self.find(y)bx,by = x,yif x == y:return Falseif self.uf[x] > self.uf[y]:x,y = y,xself.uf[x] += self.uf[y]self.uf[y] = xif bx != x:self.ps.remove(bx)if by != x:self.ps.remove(by)return Truedef size(self,x):x = self.find(x)return -self.uf[x]n,q = map(int,input().split())uf = Unionfind(n)for i in range(q):l = list(map(int,input().split()))if l[0] == 1:u,v = l[1:]uf.union(u-1,v-1)else:u = l[1]p = uf.find(u-1)ans = -1for x in uf.ps:if x != p:ans = x+1breakprint(ans)