結果
問題 | No.843 Triple Primes |
ユーザー |
![]() |
提出日時 | 2021-05-05 02:05:13 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 57 ms / 2,000 ms |
コード長 | 1,180 bytes |
コンパイル時間 | 184 ms |
コンパイル使用メモリ | 82,404 KB |
実行使用メモリ | 67,504 KB |
最終ジャッジ日時 | 2024-07-23 21:36:03 |
合計ジャッジ時間 | 3,941 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 42 |
ソースコード
import sys sys.setrecursionlimit(10**7) def I(): return int(sys.stdin.readline().rstrip()) def MI(): return map(int,sys.stdin.readline().rstrip().split()) def LI(): return list(map(int,sys.stdin.readline().rstrip().split())) def LI2(): return list(map(int,sys.stdin.readline().rstrip())) def S(): return sys.stdin.readline().rstrip() def LS(): return list(sys.stdin.readline().rstrip().split()) def LS2(): return list(sys.stdin.readline().rstrip()) def sieve_of_eratosthenes(n): # n以下の素数の全列挙 prime_list = [] is_prime = [1]*(n+1) # A[i] = iが素数なら1,その他は0 is_prime[0] = is_prime[1] = 0 for i in range(2,int(n**.5)+1): if is_prime[i]: prime_list.append(i) for j in range(i**2,n+1,i): is_prime[j] = 0 for i in range(int(n**.5)+1,n+1): if is_prime[i] == 1: prime_list.append(i) return is_prime N = I() is_prime = sieve_of_eratosthenes(N) if N == 1: print(0) exit() ans = 1 a = 0 for r in range(3,N+1): if is_prime[r]: p = r**2-2 if p > N: break if is_prime[p]: a += 1 ans += 2*a print(ans)