204. Count Primes

问题描述

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

Credits:
Special thanks to @mithmatt for adding this problem and creating all test cases.

题目链接：

---

思路分析

判断小于n的数里有多少个素数。

小于2的数中的素数个数为0，直接返回；然后设置一个bool型向量prime，长度为n，正好最大的数是n - 1，并将所有的元素都置1，表示已开始我们认为所有的数都是素数；然后开始遍历，因为只要一个数不是小于它开方后的所有素数的倍数，那它就是素数，所以我们计算所有开放后的素数的乘积，将这些数载vector中置false；最后count数组中true的个数就是最终结果。

第一个循环加判断找小于sqrt的素数，第二个循环将素数倍数的数置false。

代码

class Solution {
public:
    int countPrimes(int n) {
        if (n <= 2)
            return 0;

        vector<bool> prime(n, true);
        prime[0] = false;
        prime[1] = false;

        for (int i = 2; i < sqrt(n); i++){
            if(prime[i]){
                for (int j = 2; j * i < n; j++){
                    prime[i * j] = false;
                }
            }

        }

        int count = 0;
        for (int i = 0; i < prime.size(); i++){
            if(prime[i])
                count++;
        }
        return count;
    }
};

时间复杂度：$？？$
空间复杂度：$O(n)$

---

反思

数学不好就是很难受啊，自己做就很难想到正确的处理方法。