LeetCode 204. Count Primes(java)

Description:

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

解法一：每次判断i是不是prime，然后用一个dp[]数组来存结果，然后由于每次都要判断prime的时间太长，所以被TLE了。时间复杂度为O(n^2)，空间复杂度为O(n)。

public int countPrimes(int n) {
        if (n == 0) return 0;
        if (n == 1) return 0;
        if (n == 2) return 0;
        int[] dp = new int[n + 1];
        for (int i = 3; i <= n; i++) {
            if (isPrime(i - 1)) {
                dp[i] = dp[i-1] + 1;
            } else {
                dp[i] = dp[i-1];
            }
        }
        return dp[n];
    }
    public boolean isPrime(int n) {
        for (int i = 2; i < n; i++) {
            if (n % i == 0) return false;
        }
        return true;
    } 

解法二：用一个dp数组，首先把0~n中不是prime的全部标出来，然后遍历一遍，遇到false的就count++，最后返回count就行。时间复杂度为O(n)，（根号n*根号n），空间复杂度为O(n)。

public int countPrimes(int n) {
        if (n < 3) return 0;
        if (n == 3) return 1;
        boolean [] notPrime = new boolean[n + 1];
        for (int i = 2; i < Math.sqrt(n); i++) {
            for (int j = 2; i * j < n; j++) {
                notPrime[i * j] = true;
            }
        }
        int count  = 0;
        for (int i = 2; i < n; i++) {
            if (!notPrime[i]) count++;
        }
        return count;
    }