結果
問題 | No.106 素数が嫌い!2 |
ユーザー | maspy |
提出日時 | 2023-05-12 21:17:02 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 555 ms / 5,000 ms |
コード長 | 589 bytes |
最終ジャッジ日時 | 2023-08-19 03:36:27 |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines import numpy as np N,K = map(int,read().split()) 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] U = 2 * 10 ** 6 + 10 _,primes = make_prime(U) cnt = np.zeros(U,np.int32) for p in primes: cnt[p::p] += 1 x = cnt[2:N+1] answer = np.count_nonzero(x >= K) print(answer)