結果

問題 No.843 Triple Primes
コンテスト
ユーザー kohei2019
提出日時 2022-04-23 00:07:20
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 160 ms / 2,000 ms
コード長 665 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 155 ms
コンパイル使用メモリ 85,296 KB
実行使用メモリ 73,280 KB
最終ジャッジ日時 2026-03-10 04:32:13
合計ジャッジ時間 4,860 ms
ジャッジサーバーID
(参考情報)
judge2_1 / judge1_0
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

def primeset(N): #N以下の素数をsetで求める.エラトステネスの篩O(√Nlog(N))
    lsx = [1]*(N+1)
    for i in range(2,int(-(-N**0.5//1))+1):
        if lsx[i] == 1:
            for j in range(i,N//i+1):
                lsx[j*i] = 0
    setprime = set()
    for i in range(2,N+1):
        if lsx[i] == 1:
            setprime.add(i)
    return setprime
    
N = int(input())
if N == 1:
    print(0)
    exit()
setp = primeset(N)
lsp = list(setp)
lsp.sort()
P =len(lsp)
maxp = lsp[-1]
cnt = 0
for k in range(P):
    if lsp[k] ** 2 > maxp*2:
        break
    for j in range(P):
        if (lsp[k]**2 - lsp[j]) in setp:
            cnt += 1
print(cnt)
0