結果

問題 No.1058 素敵な数
ユーザー yuly3yuly3
提出日時 2020-07-15 12:51:17
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 833 bytes
コンパイル時間 341 ms
コンパイル使用メモリ 82,188 KB
実行使用メモリ 80,604 KB
最終ジャッジ日時 2024-05-01 12:02:44
合計ジャッジ時間 1,459 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,224 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from bisect import bisect_left

sys.setrecursionlimit(10 ** 7)
rl = sys.stdin.readline


def eratosthenes(n):
    prime = [2]
    if n == 2:
        return prime
    limit = int(n ** 0.5)
    data = [i + 1 for i in range(2, n, 2)]
    while True:
        p = data[0]
        if limit <= p:
            return prime + data
        prime.append(p)
        data = [e for e in data if e % p != 0]


def solve():
    N = int(rl())
    
    if N == 1:
        print(1)
        return
    
    prime_nums = eratosthenes(10 ** 5 + 1000)
    idx = bisect_left(prime_nums, 10 ** 5)
    p = prime_nums[idx:idx + (N + 1)]
    nice_nums = []
    for i, pi in enumerate(p[:-1]):
        for pj in p[i + 1:]:
            nice_nums.append(pi * pj)
    nice_nums.sort()
    print(nice_nums[N - 1])


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