結果

問題 No.843 Triple Primes
コンテスト
ユーザー rlangevin
提出日時 2023-01-03 10:47:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 65 ms / 2,000 ms
コード長 574 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 82,200 KB
実行使用メモリ 69,888 KB
最終ジャッジ日時 2024-11-27 01:47:50
合計ジャッジ時間 3,570 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

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())
if N == 1:
    print(0)
    exit()
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