結果

問題 No.843 Triple Primes
コンテスト
ユーザー tktk_snsn
提出日時 2021-04-24 00:48:11
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 685 bytes
コンパイル時間 130 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 32,488 KB
最終ジャッジ日時 2024-07-04 08:51:59
合計ジャッジ時間 26,409 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 37 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import chain


def prime_set(N):
    """
    Nまでの素数のsetを返す
    """
    if N < 4:
        return ({}, {}, {2}, {2, 3})[N]
    Nsq = int(N ** 0.5 + 0.5) + 1
    primes = {2, 3} | set(chain(range(5, N + 1, 6), range(7, N + 1, 6)))
    for i in range(5, Nsq, 2):
        if i in primes:
            primes -= set(range(i * i, N + 1, i * 2))
    return primes


N = int(input())
R = []
prime = prime_set(N)
for p in prime:
    if p * p < N:
        R.append(p*p)

ans = 0
for p in prime:
    for r in R:
        q = r - p
        if p <= q and q in prime:
            if p == q:
                ans += 1
            else:
                ans += 2
print(ans)
0