Python實現"求眾數"的三種方法
阿新 • • 發佈:2018-12-09
給定一個長度為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]