結果

問題 No.1451 集団登校
ユーザー 小野寺健
提出日時 2021-05-03 10:09:55
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,207 bytes
コンパイル時間 87 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 16,632 KB
最終ジャッジ日時 2024-07-21 22:59:09
合計ジャッジ時間 7,785 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 9 WA * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

def modinv(a, m):
    b, u, v = m, 1, 0
    while b != 0:
        t = a // b
        a -= t * b
        a, b = b, a
        u -= t * v
        u, v = v, u
    u %= m
    if u < 0:
        u += m
    return u

par = []
mem = []
prb = []

def init(n):
    for i in range(n):
        par.append(i)
        mem.append(1)
        prb.append(1)
        
def find(x):
    if par[x] == x:
        return x
    else:
        par[x] = find(par[x])
        return par[x]
    
def unite(x, y):
    x = find(x)
    y = find(y)
    if x == y:
        return
    if mem[x] > mem[y]:
        par[y] = x
        mem[x] += mem[y]
        mem[y] = 0
        prb[y] = 0
    else:
        par[x] = y
        if mem[x] == mem[y]:
            prb[y] += prb[x]
        else:
            prb[x] = 0
        mem[y] += mem[x]
            
def prob(x):
    if prb[x] == 0:
        return 0
    if par[x] == x:
        return prb[x]
    else:
        return prob(par[x])

N, M = map(int, input().split())

init(N)

for _ in range(M):
    a, b = map(int, input().split())
    unite(a-1, b-1)
    
    
MOD = 10**9+7
    
for i in range(N):
    p = prob(i)
    if p == 0 or p == 1:
        print(p)
    else:
        print(modinv(p, MOD))
0