結果
問題 | No.1420 国勢調査 (Easy) |
ユーザー |
![]() |
提出日時 | 2023-05-24 20:42:02 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 849 ms / 2,000 ms |
コード長 | 2,812 bytes |
コンパイル時間 | 430 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 21,420 KB |
最終ジャッジ日時 | 2024-12-23 21:01:11 |
合計ジャッジ時間 | 15,175 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
import sysreadline=sys.stdin.readlineclass UnionFind:def __init__(self,N,label=None,f=None,weighted=False):self.N=Nself.parents=[None]*self.Nself.size=[1]*self.Nself.roots={i for i in range(self.N)}self.label=labelif self.label!=None:self.label=[x for x in label]self.f=fself.weighted=weightedif self.weighted:self.weight=[0]*self.Ndef Find(self,x):stack=[]while self.parents[x]!=None:stack.append(x)x=self.parents[x]if self.weighted:w=0for y in stack[::-1]:self.parents[y]=xw^=self.weight[y]self.weight[y]=welse:for y in stack[::-1]:self.parents[y]=xreturn xdef Union(self,x,y,w=None):root_x=self.Find(x)root_y=self.Find(y)if root_x==root_y:if self.weighted:if self.weight[y]^self.weight[x]==w:return Trueelse:return Falseelse:if self.size[root_x]<self.size[root_y]:x,y=y,xroot_x,root_y=root_y,root_xself.parents[root_y]=root_xself.size[root_x]+=self.size[root_y]self.roots.remove(root_y)if self.label!=None:self.label[root_x]=self.f(self.label[root_x],self.label[root_y])if self.weighted:self.weight[root_y]=w^self.weight[x]^self.weight[y]def Size(self,x):return self.size[self.Find(x)]def Same(self,x,y):return self.Find(x)==self.Find(y)def Label(self,x):return self.label[self.Find(x)]def Weight(self,x,y):root_x=self.Find(x)root_y=self.Find(y)if root_x!=root_y:return Nonereturn self.weight[y]^self.weight[x]def Roots(self):return list(self.roots)def Linked_Components_Count(self):return len(self.roots)def Linked_Components(self):linked_components=defaultdict(list)for x in range(self.N):linked_components[self.Find(x)].append(x)return linked_componentsdef __str__(self):linked_components=defaultdict(list)for x in range(self.N):linked_components[self.Find(x)].append(x)return "\n".join(f"{r}: {m}" for r,m in linked_components.items())N,M=map(int,readline().split())UF=UnionFind(N,weighted=True)for m in range(M):A,B=map(int,readline().split())Y=int(readline())A-=1;B-=1if UF.Union(A,B,Y)==False:print(-1)exit()for x in range(N):print(UF.Weight(x,UF.Find(x)))