結果
問題 | No.1917 LCMST |
ユーザー |
|
提出日時 | 2022-03-01 01:08:58 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 2,094 bytes |
コンパイル時間 | 156 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 226,256 KB |
最終ジャッジ日時 | 2024-06-28 08:02:39 |
合計ジャッジ時間 | 14,315 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 1 RE * 41 |
ソースコード
class UnionFindVerSize():def __init__(self, N):self._parent = [n for n in range(0, N)]self._size = [1] * Nself.group = Ndef find_root(self, x):if self._parent[x] == x: return xself._parent[x] = self.find_root(self._parent[x])stack = [x]while self._parent[stack[-1]]!=stack[-1]:stack.append(self._parent[stack[-1]])for v in stack:self._parent[v] = stack[-1]return self._parent[x]def unite(self, x, y):gx = self.find_root(x)gy = self.find_root(y)if gx == gy: returnself.group -= 1if self._size[gx] < self._size[gy]:self._parent[gx] = gyself._size[gy] += self._size[gx]else:self._parent[gy] = gxself._size[gx] += self._size[gy]def get_size(self, x):return self._size[self.find_root(x)]def is_same_group(self, x, y):return self.find_root(x) == self.find_root(y)import sys,random,bisectfrom collections import deque,defaultdictfrom heapq import heapify,heappop,heappushfrom itertools import permutationsfrom math import log,gcdinput = lambda :sys.stdin.readline().rstrip()mi = lambda :map(int,input().split())li = lambda :list(mi())M = 5 * 10**4N = int(input())A = li()res = 0cnt = [0] * (M+1)for a in A:cnt[a] += 1val_to_idx = [-1] * (M+1)B = []for i in range(1,M+1):if cnt[i]:val_to_idx[i] = len(B)B.append(i)res += (cnt[i]-1) * icnt[i] = 1n = len(B)edge = []for val in range(1,M+1):mini = M+1for i in range(val,M+1,val):if cnt[i]:mini = ibreakif mini == M+1:continuefor i in range(val,M+1,val):if cnt[i] and i!=mini:u = val_to_idx[mini]v = val_to_idx[i]edge.append((mini*i//val,u,v))edge.sort(key=lambda e:e[0])uf = UnionFindVerSize(n)for c,u,v in edge:if not uf.is_same_group(u,v):uf.unite(u,v)res += cprint(res)