LeetCode - #229 多数元素 II

前言

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

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

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

难度水平：中等

1. 描述

给定一个大小为 n 的整数数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。

2. 示例

示例 1

输入：nums = [3,2,3]
输出：[3]

示例 2

输入：nums = [1]
输出：[1]

示例 3

输入：nums = [1,2]
输出：[1,2]

提示：

• 1 <= nums.length <= 5 * 10^4
• -10^9 <= nums[i] <= 10^9
  

3. 答案

 class MajorityElementII {
    func majorityElement(_ nums: [Int]) -> [Int] {
        var num0: Int?
        var num1: Int?
        var count0 = 0
        var count1 = 0
        var res = [Int]()
        
        for num in nums {
            if let num0 = num0, num0 == num {
                count0 += 1
            } else if let num1 = num1, num1 == num {
                count1 += 1
            } else if count0 == 0 {
                num0 = num
                count0 = 1
            } else if count1 == 0 {
                num1 = num
                count1 = 1
            } else {
                count0 -= 1
                count1 -= 1
            }
        }
        
        count0 = 0
        count1 = 0
        
        for num in nums {
            if num == num0 {
                count0 += 1
            }
            if num == num1 {
                count1 += 1
            }
        }
        
        if count0 > nums.count / 3 {
            res.append(num0!)
        }
        if count1 > nums.count / 3 {
            res.append(num1!)
        }
        
        return res
    }
}

• 主要思想：遍历数组并相应地跟踪大多数元素，不要忘记在第一次迭代后验证它们是否有效。
• 时间复杂度：O(n)
• 空间复杂度：O(1)

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

点击前往 LeetCode 练习

关于我们

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

后续还会翻译大量资料到我们公众号，有感兴趣的朋友，可以加入我们。