結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー はむ吉🐹はむ吉🐹
提出日時 2016-07-10 21:25:31
言語 PyPy2
(7.3.15)
結果
AC  
実行時間 443 ms / 1,000 ms
コード長 1,725 bytes
コンパイル時間 1,296 ms
コンパイル使用メモリ 77,160 KB
実行使用メモリ 92,448 KB
最終ジャッジ日時 2024-12-15 14:24:55
合計ジャッジ時間 6,932 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 70 ms
75,672 KB
testcase_01 AC 69 ms
75,920 KB
testcase_02 AC 71 ms
75,392 KB
testcase_03 AC 192 ms
79,472 KB
testcase_04 AC 78 ms
75,904 KB
testcase_05 AC 77 ms
75,904 KB
testcase_06 AC 73 ms
75,904 KB
testcase_07 AC 74 ms
75,796 KB
testcase_08 AC 70 ms
75,528 KB
testcase_09 AC 71 ms
76,160 KB
testcase_10 AC 72 ms
75,940 KB
testcase_11 AC 78 ms
75,648 KB
testcase_12 AC 77 ms
75,904 KB
testcase_13 AC 78 ms
75,648 KB
testcase_14 AC 77 ms
76,800 KB
testcase_15 AC 71 ms
75,544 KB
testcase_16 AC 72 ms
76,160 KB
testcase_17 AC 71 ms
75,800 KB
testcase_18 AC 71 ms
75,944 KB
testcase_19 AC 153 ms
80,096 KB
testcase_20 AC 225 ms
83,348 KB
testcase_21 AC 71 ms
75,776 KB
testcase_22 AC 68 ms
75,520 KB
testcase_23 AC 69 ms
75,648 KB
testcase_24 AC 70 ms
75,648 KB
testcase_25 AC 287 ms
84,948 KB
testcase_26 AC 69 ms
75,796 KB
testcase_27 AC 70 ms
75,648 KB
testcase_28 AC 69 ms
75,904 KB
testcase_29 AC 71 ms
76,072 KB
testcase_30 AC 71 ms
75,520 KB
testcase_31 AC 71 ms
75,716 KB
testcase_32 AC 196 ms
82,052 KB
testcase_33 AC 443 ms
92,448 KB
testcase_34 AC 424 ms
92,320 KB
testcase_35 AC 264 ms
84,676 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env pypy
# -*- coding: utf-8 -*-

# 制約再変更後の想定解
# 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 = int(math.sqrt(end))
    is_prime = array.array("B", (False for _ in xrange(end + 1)))
    for x, y in itertools.product(xrange(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 xrange(5, sl + 1):
        if is_prime[n]:
            for k in xrange(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 xrange(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 = itertools.imap(int, raw_input().split())
    assert MIN_N <= n <= MAX_N
    assert MIN_L <= l <= MAX_L
    print(count_sequences(n, l))


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