leetcode 204. Count Primes

Description

Count the number of prime numbers less than a non-negative number, n.

My solution

1. 原始方案
   比较朴素的想法就是迭代的找下去, 然后把找到的质数存起来, 对于下一个数, 如果他不能被前面任何一个质数整除, 那这个数也是一个质数(我感觉是这样的). 代码如下:

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

1. 根据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统计一下, 十分微妙.

Reference

• leetcode
• c++ count
• 埃拉托斯特尼筛法