169.多数元素
https://leetcode-cn.com/problems/majority-element-ii/solution/liang-fu-dong-hua-yan-shi-mo-er-tou-piao-fa-zui-zh/
摩根投票法
摩根投票法思路
根据上述的算法思想,我们遍历投票数组,将当前票数最多的候选人与其获得的(抵消后)票数分别存储在 major 与 count 中。
当我们遍历下一个选票时,判断当前 count 是否为零:
若 count == 0,代表当前 major 空缺,直接将当前候选人赋值给 major,并令 count++
若 count != 0,代表当前 major 的票数未被完全抵消,因此令 count–,即使用当前候选人的票数抵消 major 的票数
复杂度
时间复杂度:O(n)O(n),仅遍历一次数组
空间复杂度:O(1)O(1),没有使用额外空间
class Solution(object):
def majorityElement(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
major = 0
count = 0
for n in nums:
if count == 0:
major = n
if n == major:
count = count + 1
else:
count = count - 1
return major
从第一个数开始count=1,遇到相同的就加1,遇到不同的就减1,减到0就重新换个数开始计数,总能找到最多的那个
总结
题目中给出条件「给定的数组总是存在多数元素」,因此没有涉及「计数阶段」,仅在「对抗阶段」对候选人票数进行了「对抗抵消」。如果想加深理解,推荐大家可以再做做这道题:LeetCode 229. 求众数 II。
229.求众数
def majorityElement(self, nums: List[int]) -> int:
a, b, count_a, count_b = 0, 0, 0, 0 # 设定1号众数和2号众数
res = []
for i in nums:
if a == i: # 频数统计的优先顺序要大于频数为0的判断
count_a += 1
continue
if b == i:
count_b += 1
continue
if count_a == 0:
a = i
count_a = 1
continue
if count_b == 0:
b = i
count_b = 1
continue
count_a -= 1
count_b -= 1
count_a, count_b = 0, 0 # 重置计数器
for j in nums: # 再检验
if j == a:
count_a += 1
elif j == b:
count_b += 1
if count_a > len(nums)/3:
res.append(a)
if count_b > len(nums)/3:
res.append(b)
return res