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;
}
};