给定一个大小为 n 的数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
说明: 要求算法的时间复杂度为 O(n),空间复杂度为 O(1)。
示例 1:
输入: [3,2,3]
输出: [3]
示例 2:
输入: [1,1,1,3,3,2,2,2]
输出: [1,2]
思路: 弄了个通用模板,N/K的众数最多只有k-1个。
用摩尔投票法的思想,就是“凑够”K个不同的对象,然后把他们“移除”!投完票之后的候选者“可能”是众数。
多数投票升级版:
超过n/3的数最多只能有两个。先选出两个候选人A,B。 遍历数组,分三种情况:
1.如果投A(当前元素等于A),则A的票数++;
2.如果投B(当前元素等于B),B的票数++;
3.如果A,B都不投(即当前与A,B都不相等),那么检查此时A或B的票数是否减为0:
3.1 如果为0,则当前元素成为新的候选人;
3.2 如果A,B两个人的票数都不为0,那么A,B两个候选人的票数均减一;
遍历结束后选出了两个候选人,但是这两个候选人是否满足>n/3,还需要再遍历一遍数组,找出两个候选人的具体票数。
class Solution:
def majorityElement(self, nums: List[int]) -> List[int]:
c1 = c2 = 0
n1 = n2 = None
for i in nums:
if n1 == i:
c1 += 1
elif n2 == i:
c2 += 1
elif c1 == 0:
n1 , c1 = i ,1
elif c2 == 0:
n2 , c2 = i ,1
else:
c1 , c2 = c1-1 , c2-1
return [n for n in (n1,n2) if n is not None and nums.count(n) > len(nums)/3]
思路参考:leetcode 169 投票法
最能理解的思路 以及代码写法
class Solution:
def majorityElement(self, nums: List[int]) -> List[int]:
c1 = c2 = 0
n1 = n2 = None
for i in nums:
if n1 == i:
c1 += 1
elif n2 == i:
c2 += 1
else:
if c1 == 0:
n1 , c1 = i , 1
elif c2 == 0:
n2 , c2 = i , 1
else:
c1 ,c2 = c1-1 , c2-1
return [n for n in (n1,n2) if n is not None and nums.count(n) > len(nums)/3]