Leetcode-204. 计数质数

Leetcode-204. 计数质数

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

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

思路：

     用dp数组存放该位置上的数值是否为质数。
     第一次循环遍历每个数的时候，并且把其小于n的倍数dp位置上都置为false；统计位置上为true 的数字的个数。

C++ code：

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