统计所有小于非负整数 n 的质数的数量。
示例:
输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。
此处引入求质数的一个方法,Sieve of Eratosthenes,埃拉托斯特尼筛法。
首先我们知道2是一个素数,那么 2 × 2 = 4, 3 × 2 = 6, 4 × 2 = 8…都不可能是素数了。
然后我们发现3也是素数,那么 3 × 2 = 6, 3 × 3 = 9, 3 × 4 = 12…也都不可能是素数了。
素数的倍数不可能是素数,于是在我们确定一个素数之后 ,可以将后面所有该素数的整数倍筛除。
转换成代码如下:
public int countPrimes(int n) {
boolean[] isPrim = new boolean[n+1];
Arrays.fill(isPrim, true);//此处可省略,默认为false,改下面if判断条件可节省时间,此处为了理解填充为true
for (int i = 2; i*i < n; i++) {//此处优化,原为 for (int i = 2; i < n+1; i++)
if (isPrim[i]) {
for (int j = i * i; j < n + 1; j = j + i) {//此处优化:原为for (int j = 2 * i; j < n + 1; j = j + i)
isPrim[j] = false;
}
}
}
int count = 0;
if (n <= 2) {
return 0;
}
if (n == 3) {
return 1;
} else {
for (int i = 2; i < isPrim.length-1; i++) {
if (isPrim[i]) {
count++;
}
}
}
return count;
}
优化一:在判断是否为素数的时候,循环不用写到n,而是写到Math.sqrt(n)即可;
优化二:内层循环中,j不用每次从2\*i开始,可以从i*i开始,避免重复计算,
比如:在i = 4的时候(打个比方,这种可能性不存在),2\*4=8,3\*4=12已
经在i=2和i=3的时候计算过,因此可以从4\*4=16开始筛选。