返回数组中超过长度半数数量的元素的值
一开始想的是进行sort排序,然后取中间值与两头的分别比对一下,如果一致就返回。
class Solution:
def majorityElement(self, nums: List[int]) -> int:
nums.sort(reverse = True)
n = len(nums)
mid = int(n // 2)
if nums[n-1] == nums[mid] or nums[0] == nums[mid]:
return nums[mid]
else:
return 0
但是提交的时候发现还有这种情况:
于是调整一下思路:
因为排序后的数组,如果存在一个数超过数组长度一半,中位数一定是那个数,所以使用Counter计数,只需要判断中位数的次数有没有超过长度一半即可
class Solution:
def majorityElement(self, nums: List[int]) -> int:
nums.sort(reverse = True)
n = len(nums)
mid = int(n // 2)
counts = Counter(nums)
if counts[nums[mid]]>= mid :
return nums[mid]
else:
return 0