結果

問題 No.1917 LCMST
ユーザー shinichi
提出日時 2023-01-29 17:26:22
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 3,124 bytes
コンパイル時間 584 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 644,924 KB
最終ジャッジ日時 2024-06-29 13:45:08
合計ジャッジ時間 10,960 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 1 WA * 5 MLE * 1 -- * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import defaultdict
from math import gcd

lcm = lambda x, y: x * y // gcd(x, y)

class Eratosthenes():
    def __init__(self, size):
        self.size = size
        self.isprime = [True] * size
        self.minfactor = [-1] * size
        self.mobius = [1] * size
        self.isprime[1] = False
        self.minfactor[1] = 1
        self.eratosthenes()

    # 初めに篩にかけて,minfactorとisprimeを生成
    def eratosthenes(self):
        for p in range(2, self.size):
            if not self.isprime[p]:
                continue
            self.minfactor[p] = p
            self.mobius[p] = -1
            for q in range(p+p, self.size, p):
                self.isprime[q] = False
                if self.minfactor[q] == -1:
                    self.minfactor[q] = p
                if (q//p) % p == 0:
                    self.mobius[q] = 0
                else:
                    self.mobius[q] = -self.mobius[q]

    # 高速素因数分解
    def prime_factorize(self, number):
        assert 1 <= number <= self.size
        factors = defaultdict(int)
        while number != 1:
            factors[self.minfactor[number]] += 1
            number //= self.minfactor[number]
        return factors

    # 高速約数列挙
    def divisors(self, number):
        res = [1]
        factors = self.prime_factorize(number)
        for p, cnt in factors.items():
            # 追加前の大きさを保存
            res_size = len(res)
            for i in range(res_size):
                pp = 1
                for j in range(cnt):
                    pp *= p
                    res.append(res[i]*pp)
        return res

class UnionFind(object):
    def __init__(self, size):
        self.parent = [-1] * size

    def union(self, x, y):
        x, y = self.root(x), self.root(y)
        if x == y:
            return False
        if self.parent[x] > self.parent[y]:
            x, y = y, x
        self.parent[x] += self.parent[y]
        self.parent[y] = x
        return True

    def root(self, x):
        if self.parent[x] < 0:
            return x
        self.parent[x] = self.root(self.parent[x])
        return self.parent[x]

    def same(self, x, y):
        return self.root(x) == self.root(y)

    def size(self, x):
        return - self.parent[self.root(x)]

    def group_count(self):
        return [p < 0 for p in self.parent].count(True)

N = int(input())
A = list(map(int, input().split()))

edges = []
era = Eratosthenes(2 * 10 ** 5)
minA = min(A)
minA_idx = A.index(minA)
setA = set(A)
to_idx = defaultdict(list)
for i, a in enumerate(A):
    if to_idx[a]:
        edges.append((i, to_idx[a][-1], a))
    else:
        to_idx[a].append(i)

for i, a in enumerate(A):
    divisors = era.divisors(a)
    for d in divisors:
        if to_idx[d] and to_idx[d][0] != i:
            j = to_idx[d][0]
            edges.append((i, j, lcm(A[i], A[j])))
    edges.append((i, minA_idx, lcm(a, minA)))


cost = 0
uf = UnionFind(N)
edges.sort(key= lambda x:x[-1])
for u, v, w in edges:
    if uf.same(u, v):
        continue
    uf.union(u, v)
    cost += w

print(cost)
0