(1)问题描述
给定一个数n,求出小于n的所有素数。
(2)方法一:直接判断
这样做的话,每判断一个数n,时间复杂度为O(sqrt(n))。
总的时间复杂度是O(n*(sqrt(n)))。
代码实现如下:
#include<iostream>
#include<cmath>
using namespace std;
bool is_prime(int x)
{
int sq = sqrt(x);
for(int i = 2; i <= sq; i++)
{
if(x % i == 0)
return false;
}
return true;
}
void get_prime(bool* prime, int n)
{
prime[0] = false;
prime[1] = false;
prime[2] = true;
for(int i = 3; i < n; i++)
{
prime[i] = is_prime(i);
}
}
int main()
{
int n;
cin >> n;
bool* prime = new bool[n];
get_prime(prime, n);
for(int i = 0; i < n; i++)
{
if(prime[i])
cout << i << " ";
}
cout << endl;
delete [] prime;
return 0;
}
(3)方法二:普通版素数筛法
素数筛法的核心思想是把素数的2倍、3倍、4倍...这些合数筛去。
初始化prime[0] = false,prime[1] = false,初始化其他数为素数;
2是最小的素数,所以初始化prime[2] = true,然后第一次筛选就可以筛去2的倍数,4,6,8,10,12等;
接下来到下一次筛选,发现prime[3] = true,可以筛去3的倍数,6,9,12,15等;
一直筛选下去即可得到小于n的所有素数。
这种做法的时间复杂度是O(nlog(n))。为什么是log(n)?因为对于每一个数i,假设要进行k次操作,那么就会有不等式2的k次方乘以i < n,复杂度为log(n)。
代码实现如下:
#include<iostream>
using namespace std;
void sieve(bool* prime, int n)
{
fill(prime, prime + n, 1);
prime[0] = false;
prime[1] = false;
prime[2] = true;
for(int i = 2; i < n; i++)
{
if(prime[i])
{
for(int j = 2 * i; j < n; j += i)
{
prime[j] = false;
}
}
}
}
int main()
{
int n;
cin >> n;
bool* prime = new bool[n];
sieve(prime, n);
for(int i = 0; i < n; i++)
{
if(prime[i])
cout << i << " ";
}
cout << endl;
delete [] prime;
return 0;
}
(4)方法三:快速版素数筛选
其实还可以对方法二进行优化,为什么呢?
我们仔细看就会发现,对于素数2, 我们筛去4,6,8,10,12,14,16,18等;
对于素数3,我们筛去6,9,12,15,18等;
它们重复筛去了6,12,18,而这是没有必要的(前面的已经为我筛去了某些合数,我就真的不必要轰轰烈烈地再去做一次)
那么应该怎么做呢?之前我们是这样枚举的,从这个素数的2倍开始。进行优化就应该从j = i * i开始,因为2 * i,3 * i等等这些必定被小于i的数枚举过了,已经被筛去了。
for(int j = 2 * i; j < n; j += i)
{
prime[j] = false;
}
所以上面这段代码改成这样
for(long long j = i * i; j < n; j += i)
{
prime[j] = false;
}
这种做法的时间复杂度是O(nlog(logn))。
代码实现如下:
</pre><pre>
#include<iostream>
using namespace std;
void fast_sieve(bool* prime, int n)
{
fill(prime, prime + n, 1);
prime[0] = false;
prime[1] = false;
prime[2] = true;
for(long long i = 2; i < n; i++)
{
if(prime[i])
{
for(long long j = i * i; j < n; j += i)
{
prime[j] = false;
}
}
}
}
int main()
{
int n;
cin >> n;
bool* prime = new bool[n];
fast_sieve(prime, n);
for(int i = 0; i < n; i++)
{
if(prime[i])
cout << i << " ";
}
cout << endl;
delete [] prime;
return 0;
}
这里有一个问题需要注意:fast_sieve()函数里i和j要声明为long long,因为有i * i,如果只声明为int,会溢出,导致程序运行崩溃。