题目
给定一个大小为 n 的整数数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
示例:
输入:[3,2,3]
输出:[3]
输入:[1,1,1,3,3,2,2,2]
输出:[1,2]
分析
-
遍历统计法
遍历整个数组,统计每个数的出现次数,返回频次大于 n/3 的数字组成的list。
时间复杂度:O(n)
空间复杂度:O(n) -
摩尔投票法
基本摩尔投票问题:找出一组数字序列中出现次数大于总数1/2的数(并假定该数一定存在)。
利用反证法我们可以证明,若存在这样的数字,该数字唯一。基于此:每次从序列中选择两个不同的数字删除掉,最后剩下的一个数字或几个相同的数字就是目标数。
例如,如下图所示,每种颜色的圆表示一个数,圆的数量表示该数出现的次数,按照上述步骤进行抵消,最差的情况为每次抵消都在出现频次最高的数和其他数之间,活到最后的数就是出现次数大于n/2的数。
将基本摩尔投票问题进行推广,我们可以得出:在一组数字序列中,出现次数大于n/k的数最多有k-1个。此时怎么做抵消呢?
用list存储k-1个数及其频次,在遍历数字序列的过程中更新list,假设当前遍历元素为x:- 若x在list中,将其频次+1
- 若x不在list中,则检查list中是否有频次为0的元素
- 若有,则用x代替之,并将x的频次设为1
- 若没有,则让list中所有元素的频次-1
遍历完数字序列后,从list中筛选出频次符合条件的元素返回,因为符合条件的元素不一定有k-1个,k-1是可能的最大元素数。
时间复杂度:O(n)
空间复杂度:O(1),只需要常数个元素用来存储关键元素和统计次数即可。
实现
统计遍历法:
class Solution(object):
def majorityElement(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
freq = len(nums) // 3
rlist = []
num_dict = {}
for i in nums:
if(i in rlist):
continue
if(i in num_dict.keys()):
num_dict[i] += 1
else:
num_dict[i] = 1
if(num_dict[i] > freq):
rlist.append(i)
return(rlist)
摩尔投票法 ps:该方法实际运行时间要略长于上面的方法,内存消耗比上面少0.1MB
class Solution(object):
def majorityElement(self, nums):
"""
:type nums: List[int]
:rtype: 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