[leetcode] Count Primes

Count Primes

Description:

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

click to show more hints.

References:

How Many Primes Are There?

Sieve of Eratosthenes

Credits:
Special thanks to @mithmatt for adding this problem and creating all test cases.

Have you met this question in a real interview?

 

class Solution
{
public:
  int countPrimes(int n)
  {
    vector<int> a(n);
    for(int i=; i<n; i++)
      a[i] = i;
 
    for(int i=; i * <= n; i++)
    {
      if(i == )
      {
        a[i] = ;
        continue;
      }
 
      for(int j = i + i; j < n; j += i)
      {
        if(a[j] != )
          a[j] = ;
      }
    }
 
    int count = ;
    for(int i=; i<n; i++)
    {
      cout << a[i] << " ";
      if(a[i] != )
      {
        count++;
      }
    }
    cout << endl;
 
    return count;
  }
};