【HWOJ】Prime Plus—Python

题目描述

输入一个大于0的整数，输出小于等于输入整数的质数个数

输入大于0小于100000001

输出 素数的个数

样例 输入6 输出 3

————————————————————

原题里有说明什么是素数，即一个自然数，只有两个不同的自然数约数，1和它自身，例如，前四个素数是2，3，5，7。这里要注意，1不是素数

了解了题目意思，我们就可以通过遍历小于等于输入n的自然数，判断其是不是素数，然后统计素数个数，来得到答案

def is_prime(n):
    """判断一个数是否为质数"""
    if n < 2:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True

def count_primes(n):
    """计算小于等于n的质数个数"""
    count = 0
    for i in range(2, n + 1):
        if is_prime(i):
            count += 1
    return count

# 读取输入
n = int(input("请输入一个大于0小于100000001的整数: "))

# 检查输入是否符合要求
if n <= 0 or n >= 100000001:
    print("输入的数字不符合要求，请输入一个大于0小于100000001的整数。")
else:
    # 输出素数的个数
    print(count_primes(n))

但是这段代码运行时间超过了限制，为了提高代码效率，进行了优化，不过里面的数学原理是没有问题的（好多同学是被数学或者语文难住了，而不是编程），判断一个数是不是质数，只需要判断它是否能被2到其平方根整除即可，原理自行百度

下面是优化后的代码

def sieve_of_eratosthenes(n):
    """使用埃拉托斯特尼筛法计算小于等于n的质数个数"""
    if n < 2:
        return 0
    primes = [True] * (n + 1)
    primes[0] = primes[1] = False
    for i in range(2, int(n**0.5) + 1):
        if primes[i]:
            for j in range(i * i, n + 1, i):
                primes[j] = False
    return sum(primes) 

# 读取输入
n = int(input())

print(sieve_of_eratosthenes(n))