結果
| 問題 |
No.843 Triple Primes
|
| コンテスト | |
| ユーザー |
rlangevin
|
| 提出日時 | 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 |
ソースコード
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)
rlangevin