LeetCode - #204 计数质数

持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第5天，点击查看活动详情

前言

我们社区陆续会将顾毅（Netflix 增长黑客，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新到 203 期，我们会保持更新时间和进度（周一、周三、周五早上 9:00 发布），每期的内容不多，我们希望大家可以在上班路上阅读，长久积累会有很大提升。

不积跬步，无以至千里；不积小流，无以成江海，Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言，我们会尽力满足大家的需求。

难度水平：中等

1. 描述

给定整数 n ，返回 所有小于非负整数 n 的质数的数量 。

2. 示例

示例 1

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

示例 2

输入：n = 0
输出：0

示例 3

输入：n = 1
输出：0

提示：

• 0 <= n <= 5 * 10^6

3. 答案

class CountPrimes {
    func countPrimes(_ n: Int) -> Int {
        guard n > 2 else {
            return 0
        }
      
        // init isPrime bool array
        var isPrime = Array(repeating: true, count: n)
        isPrime[0] = false
        isPrime[1] = false
        
        // count prime number
        var count = 0
        for num in 2..<n {
            guard isPrime[num] else {
                continue
            }
            
            count += 1
            
            var nextNum = num * num
            while nextNum < n {
                isPrime[nextNum] = false
                nextNum += num
            }
        }
        
        return count
    }
}

• 主要思想：创建一个布尔数组来确定质数或不是质数，筛选器的数字是从它的平方开始的前一个数字的倍数。
• 时间复杂度： O(n)
• 空间复杂度： O(n)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护，我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容，也整理收集优秀的学习资料。