【leetcode--169.多数元素】

路有瑶台2024-06-02 18:12

返回数组中超过长度半数数量的元素的值

一开始想的是进行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