[Golang]力扣Leetcode—初级算法—数学—计数质数(厄拉多塞筛法)

  • Post author:
  • Post category:golang


[Golang]力扣Leetcode—初级算法—数学—计数质数(厄拉多塞筛法)


题目

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


链接



力扣Leetcode—初级算法—数学—计数质数

.


示例1

输入:n = 10

输出:4

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


示例2

输入:n = 0

输出:0


示例3

输入:n = 1

输出:0


标签

:数组、数学、枚举、数论


思路

:如果用暴力破解,那么毫无疑问会超时,我们可以用厄拉多塞筛法(Eratosthenes):即从2开始,2的倍数全部去掉;接下来3是质数,3的倍数全部去掉,如此下去,留下的都是质数。

  • 我们首先定义一个长度为n的bool数组,用来标识某个数是否是素数,如果不是素数就将数组中对应的位置置为true
  • 首先,0和1肯定不是质数,所以数组前两个元素的值置为true
  • 数组中初始化的时候每个元素对应的值都是false,从2开始,如果对应数组中的值还是false,那么说明了这个数一定是质数
  • 所以2是质数,那么将2翻倍的数一定不是质数,所以我们将2一直这样翻倍翻下去,直到某一个倍数超过n,那么数组中就没有对应的位置了,我们在这个过程中,将翻倍得到的数字在数组中对应的位置置为true
  • 然后遍历2的下一个数,首先看这个数在数组中对应位置的值是true还是false,如果是true说明了这个位置是由前面某个数翻倍得到的,说明他是合数,所以跳过,如果是false,说明了这个数不能由前面的数翻倍得到,那么他肯定是质数
  • 一直重复上述过程,直到这个2变成了最后的n


主要Go代码如下:

package main

import "fmt"

func countPrimes(n int) int {
	list := make([]bool, n)
	sum := 0

	for i := 2; i < n; i++ {
		if list[i] == false {
			sum++
		}
		for j := i + i; j < n; j += i {
			list[j] = true
		}
	}
	return sum
}

func main() {
	var n int
	fmt.Scanf("%d", &n)
	fmt.Println(countPrimes(n))
}


提交截图



在这里插入图片描述



版权声明:本文为a6661314原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。