题目:
Description:
Count the number of prime numbers less than a non-negative number, n.
思路:
最对口的算法叫做Sieve of Eratosthenes筛选法,就是如何快速地筛选出某个范围内的所有素数。关于这个算法的细节读者可以参考维基百科,也可以直接通过阅读下面的代码理解。我们这里着重解释一下在对该算法的实现中,可以优化的几个地方:
1)在循环的过程中,只需要让i从2循环到sqrt(n)而不是n。这是因为,一旦n含有小于等于sqrt(n)的因数,那么必然存在大于等于sqrt(n)的因数,所以i一定不是素数。
2)当我们发现i是素数时,我们只需要从i*i开始,删除它的小于n的倍数,而不需要从0开始,这是因为比i*i小的素数在之前的循环中都已经被筛选出来了。
3)我们每次让j自增i,而不是比j小的其它素数,这是因为质因数中含有比i小的素数的合数在之前的循环中都被筛选出来了,所以不需要重复筛选。
代码:
class Solution {
public:
int countPrimes(int n) {
if(n < 1) {
return 0;
}
vector<bool> hash(n, true);
for(int i = 2; i * i < n; ++i) { // optimization 1
if(!hash[i]) {
continue;
}
int square = i * i; // optimization 2
for(int j = 0; square + j < n; j += i) { // optimization 3
hash[square + j] = false;
}
}
int ret = 0;
for(int i = 2; i < n; ++i) {
ret += hash[i];
}
return ret;
}
};