結果

問題 No.1451 集団登校
ユーザー titia
提出日時 2025-04-10 01:06:05
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,629 bytes
コンパイル時間 307 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 24,320 KB
最終ジャッジ日時 2025-04-10 01:06:18
合計ジャッジ時間 10,789 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 18 WA * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

mod=10**9+7

n,m=map(int,input().split())

class Union_find():
    def __init__(self, N): # N要素のUnion_find. A.Union_find(N)で作る.
        self.Group = [i for i in range(N)]
        self.Nodes = [1]*(N)
        
    def find(self,x): # find(a)=find(b)のとき同じグループ
        while self.Group[x] != x:
            x = self.Group[x]
        return x

    def Union(self,x,y):
        if self.find(x) != self.find(y):
            if self.Nodes[self.find(x)] < self.Nodes[self.find(y)]:
                
                self.Nodes[self.find(y)] += self.Nodes[self.find(x)]
                self.Nodes[self.find(x)] = 0
                self.Group[self.find(x)] = self.find(y)
                
            else:
                self.Nodes[self.find(x)] += self.Nodes[self.find(y)]
                self.Nodes[self.find(y)] = 0
                self.Group[self.find(y)] = self.find(x)


uf1=Union_find(n)
uf2=Union_find(n)

TO=[i for i in range(n)]

for i in range(m):
    x,y=map(int,input().split())

    x-=1
    y-=1

    k=uf1.find(x)
    l=uf1.find(y)

    to_k=TO[k]
    to_l=TO[l]

    if k==l:
        continue

    if uf1.Nodes[k]<uf1.Nodes[l]:
        uf1.Union(k,l)
        TO[uf1.find(x)]=to_l
    elif uf1.Nodes[l]<uf1.Nodes[k]:
        uf1.Union(k,l)
        TO[uf1.find(x)]=to_k
    else:
        uf2.Union(to_k,to_l)
        uf1.Union(k,l)
        TO[uf1.find(x)]=uf2.find(to_k)

ANS=[0]*n

for i in range(n):
    x=uf1.find(i)
    to=TO[x]

    if uf2.find(i)==to:
        nx=uf2.Nodes[uf2.find(i)]

        print(pow(nx,mod-2,mod))

    else:
        print(0)
0