leetcode 204.计数质数 Java

题目链接

https://leetcode-cn.com/problems/count-primes/

描述

统计所有小于非负整数 n 的质数的数量。

提示：

0 <= n <= 5 * 106

示例

示例 1：

输入：n = 10
输出：4
解释：小于 10 的质数一共有 4 个, 它们是 2, 3, 5, 7 。

示例 2：

输入：n = 0
输出：0

示例 3：

输入：n = 1
输出：0

初始代码模板

class Solution {
    public int countPrimes(int n) {

    }
}

代码

推荐官方的题解：
https://leetcode-cn.com/problems/count-primes/solution/ji-shu-zhi-shu-by-leetcode-solution/

class Solution {
    public int countPrimes(int n) {
        boolean[] nums = new boolean[n];
        int ans = 0;
        for (int i = 2; i < n; i++) {
            if (!nums[i]) {
                ans += 1;

                if ((long)i * i >= n) {
                    continue;
                }
                for (int j = i * i; j < n; j += i) {
                    nums[j] = true;
                }
            }
            //System.out.println(Arrays.toString(nums));
        }

        return ans;
    }

}