前言
做题时碰到一个需要所有素数的一道题 这篇讲一下怎么筛选出来所有素数
埃拉托斯特尼筛法
核心思想是利用质数的倍数来标记非质数
代码如下:
// 定义全局变量
int prime[N]; // 存储质数的数组
int cnt = 0; // 质数的计数器
bool st[N]; // 布尔数组,用于标记非质数
// 生成不超过 n 的所有质数
void get_prime(int n)
{
// 遍历从 2 到 n 的每一个整数
for(int i = 2; i <= n; i++)
{
// 如果 st[i] 为 false,表示 i 是质数
if(!st[i])
{
// 将质数 i 存入 prime 数组
prime[cnt++] = i;
}
// 遍历所有已经发现的质数
for(int j = 0; prime[j] <= n / i; j++)
{
// 将 i 乘以 prime[j] 标记为非质数
st[prime[j] * i] = true;
// 如果 i 能被 prime[j] 整除,说明 i 之后的数都不可能是质数
if(i % prime[j] == 0)
break;
}
}
}
解释注释
1.变量定义:
int prime[N]:用于存储找到的质数。
int cnt:质数的计数器,初始值为 0。
bool st[N]:布尔数组,表示每个数是否为非质数(true 表示非质数,false 表示质数)。
int sum[N]:备用数组,在此函数中并未使用。
2.get_prime(int n) 函数:
外层循环:遍历从 2 到 n 的每一个整数 i。
如果 st[i] 为 false,说明 i 是质数,将 i 存入 prime 数组并增加计数器 cnt。
内层循环:对所有已经发现的质数 prime[j] 进行操作。
如果 prime[j] * i 小于等于 n,则将 prime[j] * i 标记为非质数。
如果 i 能被 prime[j] 整除,则跳出内层循环,因为 i 之后的数也不能是质数了。
总结
埃拉托斯特尼筛法具有 𝑂(𝑛loglog𝑛) 的时间复杂度,是找出所有小于等于 n 的质数的经典算法。