LeetCode 204. 计数质数

目录结构

1.题目

2.题解


1.题目

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

示例:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-primes
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2.题解

筛选法.

开辟一个大小为n的数组nums,默认值均为0。将nums[0]和nums[1]赋值1(0和1不是质数,这里以0代表质数,1代表合数,可省去初始化数组全部为1的过程)。

  • 从2开始,将2的倍数(不含2)4、6、8......赋值1;
  • 从3开始,将3的倍数(不含3)6、9、12......赋值1;
  • 直至n-1。

最后遍历数组统计值为0的个数即可。

public class Solution204 {
    public int countPrimes(int n) {
        if (n == 0 || n == 1 || n == 2) {
            return 0;
        }
        int[] nums = new int[n];
        nums[0] = nums[1] = 1;
        for (int i = 2; i < Math.sqrt(n); i++) {
            for (int j = 2; i * j < n; ++j) {
                nums[i * j] = 1;
            }
        }
        int count = 0;
        for (int t : nums) {
            count += t == 0 ? 1 : 0;
        }
        return count;
    }
}
  • 时间复杂度:O(n^2)
  • 空间复杂度:O(n)

你可能感兴趣的:(LeetCode,leetcode)