给定整数 n ,返回 所有小于非负整数 n 的质数的数量 。
示例 1:
输入:n = 10
输出:4
解释:小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。
示例 2:
输入:n = 0
输出:0
示例 3:
输入:n = 1
输出:0
提示:
0 <= n <= 5 * 106
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-primes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution:
def countPrimes(self, n: int) -> int:
if n < 2:
return 0
# 初始化列表且初始值为1,代表为质数
isPri = [1] * n
# 0、1为合数,不用判断
isPri[0] = isPri[1] = 0
# 遍历所给整数区间的一半(右一半为左一半的倍数,所以不考虑),若某数为合数,则其倍数也为合数
for i in range(2, int(n ** 0.5 + 1)):
# 拿到某数的所有倍数并将其标记为合数
isPri[i * i: n: i] = [0] * len(isPri[i * i: n: i])
return isPri.count(1)