結果

問題 No.1917 LCMST
コンテスト
ユーザー ああいい
提出日時 2022-04-29 23:35:07
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 962 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 82,172 KB
実行使用メモリ 839,180 KB
最終ジャッジ日時 2024-06-29 05:21:53
合計ジャッジ時間 14,081 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 6 MLE * 1 -- * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
A = list(map(int,input().split()))
A.sort()
C = 10 ** 5 + 1
dat = [-1] * C

parent = list(range(N))
import sys
sys.setrecursionlimit(10 ** 8)
def find(i):
    if parent[i] == i:
        return i
    parent[i] = find(parent[i])
    return parent[i]
def unit(i,j):
    I = find(i)
    J = find(j)
    if I == J:return False
    parent[I] = J
    parent[i] = J
    return True
def isSame(i,j):
    return find(i) == find(j)

def lcm(a,b):
    x,y = a,b
    while True:
        r = a % b
        a = b
        b = r
        if r == 0:
            return x // a * y

l = []
ans = 0
for i in range(N):
    a = A[i]
    if dat[a] >= 0:
        ans += a
        unit(i,dat[a])
        continue
    for j in range(a,C,a):
        dat[j] = i
    for j in range(i + 1,N):
        if A[j] % a == 0:continue
        l.append((lcm(a,A[j]),i,j))
l.sort(key = lambda x:x[0])
for t,i,j in l:
    if not isSame(i,j):
        ans += t
        unit(i,j)
print(ans)
0