結果

問題 No.843 Triple Primes
ユーザー rlangevinrlangevin
提出日時 2023-01-03 10:46:56
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 539 bytes
コンパイル時間 377 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 69,888 KB
最終ジャッジ日時 2024-11-27 01:47:45
合計ジャッジ時間 4,266 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 40 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff #

from math import *

def Sieve(n):
    lst = [True] * (n + 1)
    S = set()
    for i in range(2, ceil(sqrt(n)) + 1):
        if lst[i]:
            for j in range(2 * i, n + 1, i):
                lst[j] = False
    for i in range(2, n + 1):
        if lst[i]:
            S.add(i)
    return S


N = int(input())
S = set()
prime_set= Sieve(N)
i = 1
while i * i <= N + 2:
    if i in prime_set:
        S.add(i * i)
    i += 1
    

ans = 1
for v in prime_set:
    if v == 2:
        continue
    if v + 2 in S:
        ans += 2
print(ans)
0