結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー はむ吉🐹はむ吉🐹
提出日時 2016-07-10 21:16:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 509 ms / 1,000 ms
コード長 1,689 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,580 KB
実行使用メモリ 88,224 KB
最終ジャッジ日時 2024-12-15 14:10:10
合計ジャッジ時間 4,527 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 35 ms
53,804 KB
testcase_01 AC 34 ms
53,940 KB
testcase_02 AC 34 ms
53,596 KB
testcase_03 AC 104 ms
78,128 KB
testcase_04 AC 36 ms
54,048 KB
testcase_05 AC 35 ms
53,196 KB
testcase_06 AC 33 ms
53,516 KB
testcase_07 AC 35 ms
53,468 KB
testcase_08 AC 37 ms
53,392 KB
testcase_09 AC 35 ms
53,728 KB
testcase_10 AC 35 ms
53,264 KB
testcase_11 AC 35 ms
52,956 KB
testcase_12 AC 37 ms
53,012 KB
testcase_13 AC 35 ms
53,040 KB
testcase_14 AC 41 ms
60,468 KB
testcase_15 AC 36 ms
53,336 KB
testcase_16 AC 35 ms
53,380 KB
testcase_17 AC 34 ms
53,780 KB
testcase_18 AC 35 ms
54,560 KB
testcase_19 AC 133 ms
78,156 KB
testcase_20 AC 207 ms
80,600 KB
testcase_21 AC 37 ms
53,532 KB
testcase_22 AC 37 ms
54,072 KB
testcase_23 AC 37 ms
54,712 KB
testcase_24 AC 37 ms
52,904 KB
testcase_25 AC 295 ms
83,596 KB
testcase_26 AC 39 ms
52,976 KB
testcase_27 AC 39 ms
52,956 KB
testcase_28 AC 38 ms
53,572 KB
testcase_29 AC 39 ms
52,832 KB
testcase_30 AC 37 ms
53,364 KB
testcase_31 AC 36 ms
53,496 KB
testcase_32 AC 194 ms
80,572 KB
testcase_33 AC 500 ms
88,156 KB
testcase_34 AC 509 ms
88,224 KB
testcase_35 AC 271 ms
82,044 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env pypy3

# 制約再変更後の想定解
# Eratosthenesの篩を用いたPyPy3解がTLEになったため、
# Atkinの篩に切り替えた

import array
import itertools
import math


MIN_N = 3
MAX_N = 10 ** 6
MIN_L = 1
MAX_L = 2 * 10 ** 7


# Atkinの篩により、end未満の素数を列挙する
# この実装では計算量O(end)
# http://www.prefield.com/algorithm/math/sieve_of_atkin.html
def sieve_of_atkin(end, typecode="L"):
    sl = math.floor(math.sqrt(end))
    is_prime = array.array("B", (False for _ in range(end + 1)))
    for x, y in itertools.product(range(1, sl + 1), repeat=2):
        n = 4 * x * x + y * y
        if n <= end and (n % 12 == 1 or n % 12 == 5):
            is_prime[n] ^= True
        n = 3 * x * x + y * y
        if n <= end and n % 12 == 7:
            is_prime[n] ^= True
        n = 3 * x * x - y * y
        if x > y and n <= end and n % 12 == 11:
            is_prime[n] ^= True
    for n in range(5, sl + 1):
        if is_prime[n]:
            for k in range(n * n, end, n * n):
                is_prime[k] = False
    if end > 2:
        is_prime[2] = True
    if end > 3:
        is_prime[3] = True
    primes = array.array(typecode)
    primes.extend(p for p in range(end) if is_prime[p])
    return primes


def count_sequences(n, l):
    def x_max(d):
        return l - (n - 1) * d
    d_max = l // (n - 1)
    if d_max < 2:
        return 0
    ds = sieve_of_atkin(d_max + 1)
    ans = sum(x_max(d) + 1 for d in ds)
    return ans


def main():
    n, l = map(int, input().split())
    assert MIN_N <= n <= MAX_N
    assert MIN_L <= l <= MAX_L
    print(count_sequences(n, l))


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