Description
Count the number of prime numbers less than a non-negative number, n.
My solution
- 原始方案
比较朴素的想法就是迭代的找下去, 然后把找到的质数存起来, 对于下一个数, 如果他不能被前面任何一个质数整除, 那这个数也是一个质数(我感觉是这样的). 代码如下:
class Solution {
public:
int countPrimes(int n) {
if (n < 2) return 0;
bool isprime;
vector<int> primes;
for (int i = 2; i < n; i++) {
isprime = true;
for (int j = 0; j < primes.size(); j++) {
if (i % primes[j] == 0) {
isprime = false;
break;
}
}
if (isprime) primes.push_back(i);
}
return primes.size();
}
};
提交结果: TLE
- 根据discuss中的方案改进如下:
class Solution {
public:
int countPrimes(int n) {
vector<bool> isprime(n, true);
int cnt = 0;
for (int i = 2; i < n; i++) {
if (isprime[i]) {
for (int j = i; i * j < n; j++) {
isprime[i * j] = false;
}
cnt++;
}
}
return cnt;
}
};
这个方法实际上是理解到一半的埃拉托斯特尼筛法, 里面仍有方法1的思想. 直观容易理解, 但提交结果仍然是TLE!!
Discuss
参考姿势:
class Solution {
public:
int countPrimes(int n) {
vector<bool> prime(n, true);
prime[0] = false, prime[1] = false;
for (int i = 0; i < sqrt(n); ++i) {
if (prime[i]) {
for (int j = i*i; j < n; j += i) {
prime[j] = false;
}
}
}
return count(prime.begin(), prime.end(), true);
}
};
仔细对照自己的实现, 和参考实现:
1. 外层for自己的i要遍历太多
2. 内层for自己的效率低
3. return不同. 自己的实现实际上对每个数字都check了一下, return一个cnt. 参考方案check了sqrt(n)就可以把所有质数都确定下来了(因为把范围内的非质数都剔除了), return形式就是把true统计一下, 十分微妙.