题目:给定整数 n
,返回 所有小于非负整数 n
的质数的数量 。
原解题思路:
定义一个标志数flag,初始值为1,再定义一个计数器k,初始值为0。首先,如果n<2,那么不可能为质数,返回0。用双层for循环对所有小于n的数进行质数判断,如果该数对大于二的值取模为0,即该数字不是质数,则将flag赋值为0。在内层循环结束后,如果flag值为1,则计数器k+1;循环结束后,返回k的值。
代码如下:
int countPrimes(int n){
int flag=1,k=0;
if(n<2) return 0;
for(int i=2;i<n;i++){
flag=1;
for(int j=2;j<=sqrt(i);j++){
if(i%j==0){
flag=0;
break;
}
}
if(flag==1) {
k++;
}
}
return k;
}
提交代码,显示超出时间限制。
分析:使用了简单的试除法来检查每个数是否为质数,这种方法对于较大的 n 来说效率很低。所以需优化判断质数的算法,通过查看题目提示3,了解到埃拉托色尼筛法。即要得到自然数n以内的全部质数,必须把不大于根号n的所有质数的倍数剔除,剩下的就是质数。
思路:创建一个bool类型的数组isPrime,大小为n,用来标记从0到n-1的每个数是否为质数。初始化数组,将所有元素设为true,表示假设所有数都是质数。因为0和1不是质数,所以isPrime[0]和isPrime[1]设为false。使用一个循环,从2开始,直到p * p(p的平方)小于n。用来找到所有小于sqrt(n)的质数,并用它们来标记它们的倍数为非质数。如果p是一个质数,那么将p的所有倍数(从p * p开始,每次+p)在isPrime数组中标记为false。再使用循环循环,从2开始,直到n-1,遍历isPrime数组。对于数组中的每个true值,我们将质数计数器count增加1。循环结束后,count变量就包含了小于n的质数数量。最后,返回这个值作为结果。
代码:
int countPrimes(int n) {
if (n < 2)
return 0;
bool isPrime[n];
for (int i = 0; i < n; i++) {
isPrime[i] = true;
}
isPrime[0] = isPrime[1] = false;
int count = 0;
for (int p = 2; p * p < n; p++) {
if (isPrime[p]) {
// 标记所有p的倍数为非质数
for (int i = p * p; i < n; i += p) {
isPrime[i] = false;
}
}
}
// 计数剩余的质数
for (int i = 2; i < n; i++) {
if (isPrime[i]) {
count++;
}
}
return count;
}