169. 多数元素

给定一个大小为 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