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