結果
問題 |
No.732 3PrimeCounting
|
ユーザー |
![]() |
提出日時 | 2020-01-02 15:41:44 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
RE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 991 bytes |
コンパイル時間 | 157 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 44,516 KB |
最終ジャッジ日時 | 2024-11-22 18:08:08 |
合計ジャッジ時間 | 47,689 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | RE * 89 |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np N = int(read()) def make_prime(U): is_prime = np.zeros(U,np.bool) is_prime[2] = 1 is_prime[3::2] = 1 M = int(U**.5)+1 for p in range(3,M,2): if is_prime[p]: is_prime[p*p::p+p] = 0 return is_prime, is_prime.nonzero()[0] is_prime, primes = make_prime(3 * N + 10) f = np.zeros(N+1,np.int64) f[primes[primes<=N]] += 1 fft = np.fft.rfft; ifft = np.fft.irfft fft_len = 1 << 19 Ff = fft(f,fft_len); Fg = Ff ** 3; g = ifft(Fg,fft_len) g = np.rint(g).astype(np.int64)[:3*N+1] x = (g * is_prime[:3*N+1]).sum() # a == b や b == c かもしれない状態になっている。うまく間引く。 h = np.zeros(N+N+1,np.int64) h[2*primes[primes<=N]] += 1 Fh = fft(h,fft_len) g = ifft(Ff * Fh, fft_len) g = np.rint(g).astype(np.int64)[:3*N+1] y = (g * is_prime[:3*N+1]).sum() answer = (x - 3 * y) // 6 print(answer)