LeetCode Count Primes

Description:

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

click to show more hints.

References:

How Many Primes Are There?

Sieve of Eratosthenes

 

上面的这两个参考内容可以看看，第二个给出了经典的解法，在Java Core讲解BitSet的时候也提到了

不过基本版本和不断改进后的版本时间差距非常大：

第一个基本版本[722ms]：

class Solution {

public:

    int countPrimes(int n) {

        vector<bool> filter(n + 1, false);

        int cnt = 0;

        for (int i=2; i<n; i++) {

            if (!filter[i]) {

                cnt++;

                for (int j=i+i; j < n; j+=i) {

                    filter[j] = true;

                }

            }

        }

        return cnt;

    }

};

这里选择了i+i作为j的起点，其实i+i是2的倍数已经被前面的循环给置位过了，同理所有小于i的数的倍数也都被置位了，因而直接从i*i开始,[619ms]

class Solution {

public:

    int countPrimes(int n) {

        vector<bool> filter(n, false); 

        int cnt = 0;

        int lmt = sqrt(n);

        for (int i=2; i<n; i++) {

            if (!filter[i]) {

                cnt++;

                if (i > lmt) continue;

                for (int j=i*i; j < n; j+=i) {

                    filter[j] = true;

                }

            }

        }

        return cnt;

    }

};

因为质数必然不是偶数，所以外层循环可以只考虑奇数的情况，这样就减少了一半计算量[224ms]

class Solution {

public:

    int countPrimes(int n) {

        if (n <= 2) return 0;

        vector<bool> filter(n, false); 

        int cnt = 1;

        int lmt = sqrt(n);

        for (int i=3; i<n; i+=2) {

            if (!filter[i]) {

                cnt++;

                if (i > lmt) continue;

                for (int j=i*i; j < n; j+=i) {

                    filter[j] = true;

                }

            }

        }

        return cnt;

    }

};

相应的内层循环也可以只考虑奇数的情况，因为偶数的情况肯定在i=2时已经被置位了[140ms]

class Solution {

public:

    int countPrimes(int n) {

        if (n <= 2) return 0;

        vector<bool> filter(n, false); 

        int cnt = 1;

        int lmt = sqrt(n);

        for (int i=3; i<n; i+=2) {

            if (!filter[i]) {

                cnt++;

                if (i > lmt) continue;

                for (int j=i*i, step = i << 1; j < n; j+=step) {

                    filter[j] = true;

                }

            }

        }

        return cnt;

    }

};

把vector换成数组又可以提升不少速度，[41ms]

class Solution {

public:

    int countPrimes(int n) {

        if (n <= 2) return 0;

        bool* filter = new bool[n];

        for (int i=0; i<n; i++) filter[i] = false;

        int cnt = 1;

        int lmt = sqrt(n);

        for (int i=3; i<n; i+=2) {

            if (!filter[i]) {

                cnt++;

                if (i > lmt) continue;

                for (int j=i*i, step = i << 1; j < n; j+=step) {

                    filter[j] = true;

                }

            }

        }

        delete[] filter;

        return cnt;

    }

};