結果
問題 | No.1917 LCMST |
ユーザー |
![]() |
提出日時 | 2022-02-28 13:40:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,999 ms / 4,000 ms |
コード長 | 1,281 bytes |
コンパイル時間 | 161 ms |
コンパイル使用メモリ | 82,300 KB |
実行使用メモリ | 319,684 KB |
最終ジャッジ日時 | 2024-06-28 07:59:34 |
合計ジャッジ時間 | 43,251 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
class UnionFind():def __init__(self, n):self.n = nself.parents = [-1] * ndef find(self, x):if self.parents[x] < 0:return xelse:self.parents[x] = self.find(self.parents[x])return self.parents[x]def union(self, x, y):x = self.find(x)y = self.find(y)if x == y:returnif self.parents[x] > self.parents[y]:x, y = y, xself.parents[x] += self.parents[y]self.parents[y] = xdef size(self, x):return -self.parents[self.find(x)]def same(self, x, y):return self.find(x) == self.find(y)def MST(n,edge):uf=UnionFind(n)ans=0for cost,u,v in edge:if not uf.same(u,v):ans+=costuf.union(u,v)return ansimport mathimport randomn=int(input())a=list(map(int,input().split()))#n=5*10**5#a=[random.randint(50000,10**5) for i in range(n)]m=10**5d=[-1]*(m+1)cnt=[0]*(m+1)ans=0for i in a:if cnt[i]==0:cnt[i]=1else:ans+=iedge=[]for i in range(1,m+1):for j in range(i,m+1,i):if cnt[j]==1:if d[i]==-1:d[i]=jelse:edge.append((d[i]*j//i,d[i],j))edge.sort(key=lambda x:x[0])ans+=MST(m+1,edge)print(ans)