LeetCode. 计数质数

题目要求：

统计所有小于非负整数 n 的质数的数量。

示例i:

输入: 10
输出: 4
解释: 小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

代码：

class Solution {
public:
    int countPrimes(int n) {
        if (n <= 2) return 0;
        vector<bool> prime(n, true);
        for(int i = 2; i*i < n; i++) {
            if(prime[i]) {
                for(int j = i*i; j < n; j += i) {
                    prime[j] = false;
                }
            }
        }
        
        int count = 0;
        for(int i = 2; i < n; i++) {
            if(prime[i]) {
                count += 1;
            }
        }
        return count;
    }
  
};

分析：

用空间换时间的思想，保证不超时