Leetcode: Count Primes

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

Algorithm: The sieve of Eratosthenes, Referring to Summary: Primes

0 and 1 are not primes!

Maintain a notPrime boolean array. Start from the first prime 2, cross out all its multiples. The next number that is not crossed out is the next prime. Continue the process until reach n.

public class Solution {
    public int countPrimes(int n) {
        if (n <= 1) return 0;
        boolean[] notPrime = new boolean[n+1];
        notPrime[0] = true;
        notPrime[1] = true;
        int count = 0;
        for (int i=2; i<n; i++) {
            if (!notPrime[i]) {
                count++;
                int mul = 2;
                while (i <= n/mul) {
                    notPrime[i*mul] = true;
                    mul++;
                }
            }
        }
        return count;
    }
}