统计所有小于非负整数 n 的质数的数量。
示例 1:
输入:n = 10
输出:4
解释:小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。示例 2:
输入:n = 0
输出:0
示例 3:输入:n = 1
输出:0
提示:
0 <= n <= 5 * 106
遍历
遍历判断[2,n]中所有整数是否为质数。
判断每个数i(i<=n)是否为质数时,只需判断[2,sqrt(i)]中每个数是否为i的因数
厄拉多塞筛法/埃氏筛
所有合数都是质数的倍数
设置数组 isPrime[i]表示数 i 是否为质数(1:是质数;0:不是质数/是合数)。for i=2从小到大遍历每个数,如果i为质数,1.质数计数+1;2.将其所有的倍数都标记为合数(除了该质数本身)。
【注意】将质数i的m倍 i*m(m>=2) 都标记为合数,这一步其实可以从该质数i的i倍 i*i 开始标记,因为质数i的m倍可视为m的i倍,当m<i时,这些数已经被标记为合数,因此无需重复标记。
class Solution {
public:
int countPrimes(int n) {
vector<int> isPrime(n, 1);
int ans = 0;
for (int i = 2; i < n; ++i) {
if (isPrime[i]) {
ans += 1;
if ((long long)i * i < n) {
for (int j = i * i; j < n; j += i) {
isPrime[j] = 0;
}
}
}
}
return ans;
}
};
复杂度
时间复杂度:O(n*log n*log n)
空间复杂度:O(n),isPrime[n]用于标记每个数是否为质数
线性筛
埃氏筛仍有重复标记的情况(例如:30既是3的10倍,又是5的6倍)
使用线性筛减少重复标记,降低时间复杂度
设置数组isPrime[n]标记n个数的是否为合数
设置数组Primes[]记录当前遍历到的所有质数
线性筛:标记所有整数的质数倍(Primes[i])为合数
埃氏筛:标记所有质数的整数倍为合数
【注意】当整数i可以整除Primes[j]时,停止对该整数的倍数的标记。因为对于任意k>j,i*Primes[k]可以被i/Primes[i]*Primes[k]标记。这种方法确保每个合数只被它的最大公约数标记。
例如:质数集:Primes[]=[2,3,5,7],i=6,Primes[j]=3,6可以整除3。
标记i*Primes[j+1]=6*5=30时,不再用6的5倍标记,而是使用10(6/3*5)的3倍标记。
class Solution {
public:
int countPrimes(int n) {
vector<int> primes;
vector<int> isPrime(n, 1);
for (int i = 2; i < n; ++i) {
if (isPrime[i]) { //Primes存储遍历到的所有质数
primes.push_back(i);
}
for (int j = 0; j < primes.size() && i * primes[j] < n; ++j) {
isPrime[i * primes[j]] = 0; //标记整数i的质数primes[j]为合数,
if (i % primes[j] == 0) { //如果i是质数的倍数,则不标记,
break;
}
}
}
return primes.size();
}
};
复杂度
时间复杂度:O(n)
空间复杂度:O(n),isPrime[n]和Primes[]