結果

問題 No.1058 素敵な数
ユーザー はむ吉🐹はむ吉🐹
提出日時 2020-05-22 21:42:46
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 50 ms / 2,000 ms
コード長 1,476 bytes
コンパイル時間 279 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 61,056 KB
最終ジャッジ日時 2024-10-05 16:27:01
合計ジャッジ時間 1,324 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,864 KB
testcase_01 AC 50 ms
61,056 KB
testcase_02 AC 50 ms
60,928 KB
testcase_03 AC 47 ms
60,928 KB
testcase_04 AC 43 ms
60,800 KB
testcase_05 AC 44 ms
60,672 KB
testcase_06 AC 45 ms
60,800 KB
testcase_07 AC 46 ms
61,056 KB
testcase_08 AC 46 ms
60,928 KB
testcase_09 AC 44 ms
61,056 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3

import array
import itertools
import math


M = 10 ** 5 + 1


def segment_sieve(begin, end, typecode="L"):
    """Enumerates the prime numbers in [begin, end).
    :param int begin: The beginning of the interval.
    :param int end: The end of the interval.
    :param str typecode: The type of the array to be returned (optional).
    :return: The array of the prime numbers.
    :rtype: :class:`array.array`
    """
    assert begin > 1
    assert begin <= end
    sqrt_end = math.ceil(math.sqrt(end))
    is_prime_small = array.array("B", (True for i in range(sqrt_end)))
    is_prime_small[0] = False
    is_prime_small[1] = False
    is_prime = array.array("B", (True for i in range(end - begin)))
    for i in range(2, sqrt_end):
        if is_prime_small[i]:
            for j in range(2 * i, sqrt_end, i):
                is_prime_small[j] = False
            for k in range(max(2, (begin + i - 1) // i) * i, end, i):
                is_prime[k - begin] = False
    primes = array.array(typecode,
                         (i for i, cond in enumerate(is_prime, begin) if cond))
    return primes


def generate(n):
    if n <= 1:
        return 1
    primes = segment_sieve(M, M + 1000)
    prods = [p * p for p in primes]
    prods.extend(p * q for p, q in itertools.combinations(primes, 2))
    prods.sort()
    return prods[n - 2]


def main():
    n = int(input())
    res = generate(n)
    print(res)


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