[LeetCode]229. 求众数 II

张小猪的家

于 2021-09-01 19:59:39 发布

阅读量112

点赞数

分类专栏： LeetCode 文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/weixin_39574469/article/details/120047680

版权

LeetCode 专栏收录该内容

114 篇文章

订阅专栏

这篇博客介绍了如何在给定整数数组中，利用中位数性质找到所有出现次数超过数组长度三分之一的元素，挑战时间复杂度为O(n)，空间复杂度为O(1)。通过实例演示和代码实现来帮助理解这个算法技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

229. 求众数 II

难度：中等

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

**进阶：**尝试设计时间复杂度为 O(n)、空间复杂度为 O(1)的算法解决此问题。

示例 1：

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

示例 2：

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

示例 3：

输入：[1,1,1,3,3,2,2,2]
输出：[1,2]

提示：

1 <= nums.length <= 5 * 104
-109 <= nums[i] <= 109

class Solution:
    def majorityElement(self, nums: List[int]) -> List[int]:
        n1 = n2 = nums[0]
        cnt1 = cnt2 = 0
        for num in nums:
            if num == n1:
                cnt1 += 1
            elif num == n2:
                cnt2 += 1
            elif cnt1 == 0:
                n1 = num
                cnt1 = 1
            elif cnt2 == 0:
                n2 = num
                cnt2 = 1
            else:
                cnt1 -= 1
                cnt2 -= 1
        
        cnt1 = cnt2 = 0
        res = []
        for num in nums:
            if num == n1:
                cnt1 += 1
            elif num == n2:
                cnt2 += 1
        if cnt1 > len(nums) // 3:
            res.append(n1)
        if n1 != n2 and cnt2 > len(nums) // 3:
            res.append(n2)
        return res