目录
题目描述:
难度简单
给定一个大小为 n
的数组 nums
,返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋
的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例 1:
输入:nums = [3,2,3] 输出:3
示例 2:
输入:nums = [2,2,1,1,1,2,2] 输出:2
第一种解法(超时):
第二种解法(字典加集合):
class Solution:
def majorityElement(self, nums: List[int]) -> int:
data1 = set(nums)
data2 = {}
for i in data1:
data2[str(i)] = 0
for num in nums:
data2[str(num)] += 1
if data2[str(num)] > len(nums) // 2:
return num
break
第三种解法(三行代码,力扣上的大神弄得):
class Solution:
def majorityElement(self, nums: List[int]) -> int:
nums.sort()
return nums[len(nums)//2]
第四种解法(摩根解法):
class Solution:
def majorityElement(self, nums: List[int]) -> int:
candidate = None
count = 0
for num in nums:
if count == 0:
candidate = num
count += 1 if num == candidate else -1
return candidate
摩尔投票算法的核心思想是抵消。遍历整个数组,维护一个候选元素和一个计数器。如果当前元素和候选元素相同,则计数器加一,否则计数器减一。当计数器为0时,更新候选元素为当前元素。由于多数元素出现的次数比其他元素出现的次数之和还要多,所以最终的候选元素一定是多数元素。
举个例子,假设数组为 [2, 2, 1, 1, 1, 2, 2]。初始时,候选元素为 None,计数器为 0。遍历到第一个元素 2 时,候选元素变为 2,计数器变为 1。遍历到第二个元素 2 时,计数器变为 2。遍历到第三个元素 1 时,计数器变为 1。遍历到第四个元素 1 时,计数器变为 0,候选元素变为 1。遍历到第五个元素 1 时,计数器变为 1。遍历到第六个元素 2 时,计数器变为 0,候选元素变为 2。遍历到最后一个元素 2 时,计数器变为 1。最终的候选元素为 2,即为多数元素。
摩尔投票算法的时间复杂度为 O(n),空间复杂度为 O(1),是一种非常高效的解决多数元素问题的方法。