1. 程式人生 > >Python實現"求眾數"的三種方法

Python實現"求眾數"的三種方法

給定一個長度為n的陣列,返回眾數。眾數是指陣列中出現次數超過n/2次的元素

假設陣列非空,眾數一定存在

Example 1:

Input: [3,2,3]
Output: 3

Example 2:

Input: [2,2,1,1,1,2,2]
Output: 2

1:字典,累記陣列中出現的各元素的次數,一旦發現超過n/2次的元素就返回該元素

def majorityElement(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums)==1:
            return nums[0]
        numDic = {}
        for i in nums:
            if numDic.has_key(i):
                numDic[i] += 1
                if numDic.get(i)>=(len(nums)+1)/2:
                    return i
            else:
                numDic[i] = 1

2:利用list.count()方法判斷(注意for迴圈中如果是訪問整個nums列表會出現“超出時間限制”的錯誤

def majorityElement(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        for i in nums[len(nums)//2:]:
            if nums.count(i)>len(nums)//2:
                return i

3:sorted(nums)[len(nums)//2]

def majorityElement(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        return sorted(nums)[len(nums)//2]