204. Count Primes

Description:

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

本以为是个求素数的大水题，想都没想直接写了下面代码，结果直接TLE。

class Solution {
public:
int countPrimes(int n) {
if(n<2)return 0;
int ans=0;
for(int i=2;i<=n;i++){
bool fla=1;
for(int j=2;j*j<=i;j++){
if(i%j==0){
fla=0;
break;
}
}
if(fla)ans++;
}
return ans;
}
};

好吧，记得看过一个筛选法求素数的方法。查了下，它的时间复杂度为nloglogn，

于是写出下面代码然后顺利通过。

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