結果
問題 |
No.1451 集団登校
|
ユーザー |
|
提出日時 | 2021-05-03 22:50:21 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,209 bytes |
コンパイル時間 | 225 ms |
コンパイル使用メモリ | 82,688 KB |
実行使用メモリ | 93,440 KB |
最終ジャッジ日時 | 2024-07-22 13:04:24 |
合計ジャッジ時間 | 7,274 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 18 WA * 9 |
ソースコード
class Unionfind: def __init__(self,n): self.uf = [-1]*n self.leader = [[i] for i in range(n)] self.prob = [1]*n def find(self,x): if self.uf[x] < 0: return x else: 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) if x == y: return False if self.uf[x] > self.uf[y]: x,y = y,x if self.uf[x] == self.uf[y]: self.prob[x] *= 500000004 self.prob[x] %= mod self.leader[x] += self.leader[y] self.uf[x] += self.uf[y] self.uf[y] = x self.leader[y] = [] self.prob[y] = 0 return True def size(self,x): x = self.find(x) return -self.uf[x] n,m = map(int,input().split()) uf = Unionfind(n) mod = 10**9+7 for i in range(m): a,b = map(int,input().split()) uf.union(a-1,b-1) ans = [0]*n for i in range(n): if uf.find(i) != i: continue for j in uf.leader[i]: ans[j] = uf.prob[i] for i in ans: print(i)