- 求众数 II
给定一个大小为 n 的整数数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
示例 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
进阶:尝试设计时间复杂度为 O(n)、空间复杂度为 O(1)的算法解决此问题。
方法一:哈希统计
思路
我们用哈希统计数组中每个元素出现的次数,设数组的长度为 nn,返回所有统计次数超过 n/3的元素
class Solution:
def majorityElement(self, nums: List[int]) -> List[int]:
cnt = {}
ans = []
for v in nums:
if v in cnt:
cnt[v] += 1
else:
cnt[v] = 1
for item in cnt.keys():
if cnt[item] > len(nums)//3:
ans.append(item)
return ans
复杂度分析
时间复杂度:O(n)O(n),其中 nn 为数组的长度。
空间复杂度:O(n)O(n),其中 nn 为数组的长度,使用哈希表需要开辟额外的空间。
class Solution:
def majorityElement(self, nums: List[int]) -> List[int]:
ans = []
element1, element2 = 0, 0
vote1, vote2 = 0, 0
for num in nums:
# 如果该元素为第一个元素,则计数加1
if vote1 > 0 and num == element1:
vote1 += 1
# 如果该元素为第二个元素,则计数加1
elif vote2 > 0 and num == element2:
vote2 += 1
# 选择第一个元素
elif vote1 == 0:
element1 = num
vote1 += 1
# 选择第二个元素
elif vote2 == 0:
element2 = num
vote2 += 1
# 如果三个元素均不相同,则相互抵消1次
else:
vote1 -= 1
vote2 -= 1
cnt1, cnt2 = 0, 0
for num in nums:
if vote1 > 0 and num == element1:
cnt1 += 1
if vote2 > 0 and num == element2:
cnt2 += 1
# 检测元素出现的次数是否满足要求
if vote1 > 0 and cnt1 > len(nums) / 3:
ans.append(element1)
if vote2 > 0 and cnt2 > len(nums) / 3:
ans.append(element2)
return ans
复杂度分析
时间复杂度:O(n)O(n),其中 nn 为数组的长度。
空间复杂度:O(1)O(1),只需要常数个元素用来存储关键元素和统计次数即可。