結果

問題 No.2756 GCD Teleporter
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-09-14 20:44:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 665 ms / 2,000 ms
コード長 3,024 bytes
コンパイル時間 286 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 166,256 KB
最終ジャッジ日時 2024-09-14 20:44:42
合計ジャッジ時間 16,884 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 46 ms
53,760 KB
testcase_01 AC 46 ms
54,144 KB
testcase_02 AC 45 ms
53,760 KB
testcase_03 AC 46 ms
54,016 KB
testcase_04 AC 316 ms
146,816 KB
testcase_05 AC 345 ms
150,504 KB
testcase_06 AC 535 ms
159,524 KB
testcase_07 AC 502 ms
157,316 KB
testcase_08 AC 468 ms
156,368 KB
testcase_09 AC 473 ms
153,884 KB
testcase_10 AC 528 ms
159,344 KB
testcase_11 AC 665 ms
162,552 KB
testcase_12 AC 323 ms
149,376 KB
testcase_13 AC 551 ms
160,328 KB
testcase_14 AC 589 ms
159,380 KB
testcase_15 AC 312 ms
148,864 KB
testcase_16 AC 406 ms
153,556 KB
testcase_17 AC 477 ms
156,468 KB
testcase_18 AC 315 ms
148,736 KB
testcase_19 AC 638 ms
161,280 KB
testcase_20 AC 411 ms
158,248 KB
testcase_21 AC 354 ms
151,488 KB
testcase_22 AC 453 ms
158,424 KB
testcase_23 AC 271 ms
147,328 KB
testcase_24 AC 386 ms
155,468 KB
testcase_25 AC 394 ms
155,276 KB
testcase_26 AC 442 ms
157,996 KB
testcase_27 AC 431 ms
157,852 KB
testcase_28 AC 385 ms
152,988 KB
testcase_29 AC 461 ms
154,828 KB
testcase_30 AC 487 ms
157,476 KB
testcase_31 AC 361 ms
151,800 KB
testcase_32 AC 282 ms
160,536 KB
testcase_33 AC 240 ms
151,144 KB
testcase_34 AC 240 ms
146,688 KB
testcase_35 AC 303 ms
154,000 KB
testcase_36 AC 325 ms
154,044 KB
testcase_37 AC 46 ms
53,888 KB
testcase_38 AC 303 ms
166,256 KB
testcase_39 AC 327 ms
153,184 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1694

from  collections import deque

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

    # osa-k法により素因数分解を実施
    max_a = max(A)
    min_primes = [i for i in range(max_a + 1)]
    for p in range(2, max_a + 1):
        if p != min_primes[p]:
            continue

        x = 2 * p
        while x <= max_a:
            if  min_primes[x] == x:
                min_primes[x] = p
            x += p
    
    prime_divisors = [[] for _ in range(max_a +1)]
    for a0 in range(2, max_a + 1):
        a = a0
        primes = set()
        while min_primes[a] != a:
            p = min_primes[a]
            primes.add(p)
            a //= p
        if a > 1:
            primes.add(a)
        prime_divisors[a0] = primes
    
    # 各i について「連結成分」を計算し、最小の数字を記録
    exists_a = [[] for _ in range(max_a + 1)]
    for i in range(N):
        a = A[i]
        exists_a[a].append(i)

    composite_id_list = [-1] * N
    composite_primes = {}
    composite_id = 0
    for i in range(N):
        if composite_id_list[i] == -1:
            queue = deque()
            queue.append((i, 0))
            composite_id_list[i] = composite_id
            while len(queue) > 0:
                v, typ = queue.popleft()
                if typ == 1:
                    a = v
                else:
                    a = A[v]

                if a == 1:
                    continue

                if typ == 1:
                    x = a
                    while x <= max_a:
                        for w in exists_a[x]:
                            if composite_id_list[w] == -1:
                                composite_id_list[w] = composite_id
                                queue.append((w, 0))
                        x += a
                else:
                    # typ == 0:
                    for p in prime_divisors[a]:
                        if p not in composite_primes:
                            composite_primes[p] = composite_id
                            queue.append((p, 1))
            composite_id += 1

    # 連結成分に所属する数字の中での最小値を計算する
    composite_id_map = {}
    for i in range(N):
        c_id = composite_id_list[i]
        if c_id not in composite_id_map:
            composite_id_map[c_id] = max_a
        
        for d in prime_divisors[A[i]]:
            composite_id_map[c_id] = min(composite_id_map[c_id], d)

    # 連結になるように対処する
    if len(composite_id_map) == 1:
        print(0)
    else:
        min_answer = 2 * len(composite_id_map)
        for c_id0, a in composite_id_map.items():
            if a == 1:
                continue

            ans = a * (len(composite_id_map) - 1)
            min_answer = min(min_answer, ans)
        print(min_answer)






    

    # どこかの連結成分に合わせるということをする


    

                


if __name__ == "__main__":
    main()
0