给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例 1:
输入:[3,2,3]
输出:3
示例 2:
输入:[2,2,1,1,1,2,2]
输出:2
进阶:
- 尝试设计时间复杂度为 O(n)、空间复杂度为 O(1) 的算法解决此问题。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/majority-element
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一解
使用投票算法:
class Solution:
def majorityElement(self, nums: List[int]) -> int:
res, count = nums[0], 0
for num in nums:
count += 1 if res == num else -1
if count == 0:
res = num
count = 1
return res
可将判断放到上面,以省去 count = 1
,代码优化如下:
class Solution:
def majorityElement(self, nums: List[int]) -> int:
res, count = nums[0], 0
for num in nums:
if count == 0:
res = num
count += 1 if res == num else -1
return res