結果

問題 No.1611 Minimum Multiple with Double Divisors
コンテスト
ユーザー Theta
提出日時 2022-11-18 18:40:29
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,202 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 69,400 KB
最終ジャッジ日時 2024-09-20 00:01:40
合計ジャッジ時間 6,883 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample -- * 2
other TLE * 1 -- * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import count


import numpy as np


def sieve_eratosthenes(n):
    primes = np.zeros(n + 1, dtype=bool)
    primes[2] = 1
    primes[3::2] = 1
    for p in range(3, int(n ** 0.5) + 1, 2):
        if primes[p]:
            primes[p * p::2 * p] = 0
    return primes


def calc_prime_factorize(num: int) -> dict[int, int]:
    if num < 2:
        raise ValueError

    divisors = {}
    for divisor in count(2):
        if divisor ** 2 > num:
            if num != 1:
                divisors[num] = 1
            break
        while num % divisor == 0 and num != 1:
            try:
                divisors[divisor] += 1
            except KeyError:
                divisors[divisor] = 1
            num //= divisor

    return divisors


def main():
    primes = np.where((sieve_eratosthenes(10**6)) == True)
    primes = primes[0]

    primes = set(primes)

    for _ in range(int(input())):
        X = int(input())
        if X == 1:
            print(2)
            continue
        if X == 6:
            print(24)
            continue
        prime_factors = calc_prime_factorize(X)
        print(X * min(set(primes) - set(prime_factors)))


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