leetcode#169. 多数元素

题目描述

给定一个大小为 n 的数组，找到其中的多数元素。多数元素是指在数组中出现次数大于 ⌊ n/2 ⌋ 的元素。

你可以假设数组是非空的，并且给定的数组总是存在多数元素。

解法一：count()函数

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        n = len(nums)/2
        # 先用集合去重
        nums_set = set(nums)
        # 遍历集合中元素计算其在原始列表中的数量
        for num in nums_set:
            c = nums.count(num)
            # 判断是否为多数元素
            if c > n:
                return num
            

时间复杂度O(N)+O(M*N)：set()遍历原始列表O(N)，count()函数遍历的复杂度O(N)，外循环O(M)。
空间复杂度：O(M)

解法二：分治思想

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        # 分治思想
        # 数组一分为二，分别求两区间内的众数，至少其中之一和完整数组众数相同
        # 不断分治直到区间内只有一个元素，即为众数本身，返回
        def findit(left, right):  # 初值可设置为默认参数传入
            # 递归结束条件：区间内只有一个元素
            if left == right:
                return nums[left]
            # 计算中间位置
            mid = (right-left)//2 + left
            le_ma = findit(left, mid) # 左区间众数
            ri_ma = findit(mid+1, right) # 右区间众数

            ## 左右子区间的众数若相等
            if le_ma == ri_ma:
                return le_ma
            else:
            ## 若不等，遍历当前完整区间计数，取出现次数多的
                le_n = sum(1 for i in range(left,right+1) if nums[i]==le_ma)
                ri_n = sum(1 for i in range(left,right+1) if nums[i]==ri_ma)
                return le_ma if le_n>ri_n else ri_ma

        return findit(0,len(nums)-1)

投票法

众数数量超过一半，一定比其他所有元素的数量多。

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        ## 投票法
        # 假定第一个元素为众数
        candidate = nums[0]
        count = 0

        # 遍历数组，与candidate相同的数count+1，否则-1
        for i in range(len(nums)):
            # count==0时将当前元素替换为candidate
            if count == 0:
                candidate = nums[i]
            count += (1 if nums[i] == candidate else -1)

        return candidate