给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例 1:
输入:[3,2,3]
输出:3
示例 2:
输入:[2,2,1,1,1,2,2]
输出:2
进阶:
尝试设计时间复杂度为 O(n)、空间复杂度为 O(1) 的算法解决此问题。
解题思路:
根据多数元素的定义可知,该元素的数目一定是大于其他元素的总和的,这道题要求空间复杂度为O(1),leetcode评论区有人举的例子非常好,类似于多方交战,兵力多的一方获胜,A有5人,B有3人,C有1人,A与B交战后A剩2人,再与C交战后剩1人(当然如果B与C先开战,A剩下的兵力会更多,这里只考虑最坏情况),所以只需要一个计数器,不论数组的元素顺序是怎样的,只要计数器归0就认为是前面的兵力交战损失了(相当于抵消,只需要看后续的兵力情况),归0的当前元素记为多数元素,直至遍历到最后,剩下的一定是多数元素。
代码:
class Solution:
def majorityElement(self, nums: List[int]) -> int:
if len(nums) == 0:
return None
count = 1
k = nums[0]
for i in range(1, len(nums)):
if nums[i] == k:
count += 1
else:
count -= 1
if count == 0:
k = nums[i]
count = 1
return k