結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー htkb
提出日時 2018-05-27 15:22:49
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 844 ms / 1,000 ms
コード長 491 bytes
コンパイル時間 382 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 181,612 KB
最終ジャッジ日時 2024-06-28 19:06:34
合計ジャッジ時間 5,043 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import sqrt
N, L = map(int, input().split())
max_interval = int(L//(N-1))+1

primes = set(range(7, max_interval, 6)).union(range(5, max_interval, 6))
if max_interval > 2:
    primes.add(2)
if max_interval > 3:
    primes.add(3)

du = primes.difference_update
for i in range(5, int(sqrt(max_interval))+1, 6):
    if i in primes:
        du(range(i*3, max_interval, i*2))
    if i+2 in primes:
        du(range(i*3+6, max_interval, i*2+4))

print((L+1)*len(primes)+(1-N)*sum(primes))
0