【LeetCode】python 多数元素(字典)

目录

题目描述:

第一种解法(超时):

第二种解法(字典加集合):

第三种解法(三行代码,力扣上的大神弄得):

第四种解法(摩根解法):


题目描述:

难度简单

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

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

示例 1:

输入:nums = [3,2,3]
输出:3

示例 2:

输入:nums = [2,2,1,1,1,2,2]
输出:2

第一种解法(超时):

第二种解法(字典加集合):


class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        data1 = set(nums)
        data2 = {}
        for i in data1:
            data2[str(i)] = 0
        for num in nums:
            data2[str(num)] += 1
            if data2[str(num)] > len(nums) // 2:
                return num
                break

第三种解法(三行代码,力扣上的大神弄得):

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        nums.sort()
        return nums[len(nums)//2]

第四种解法(摩根解法):

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        candidate = None
        count = 0
        for num in nums:
            if count == 0:
                candidate = num
            count += 1 if num == candidate else -1
        return candidate

摩尔投票算法的核心思想是抵消。遍历整个数组,维护一个候选元素和一个计数器。如果当前元素和候选元素相同,则计数器加一,否则计数器减一。当计数器为0时,更新候选元素为当前元素。由于多数元素出现的次数比其他元素出现的次数之和还要多,所以最终的候选元素一定是多数元素。

举个例子,假设数组为 [2, 2, 1, 1, 1, 2, 2]。初始时,候选元素为 None,计数器为 0。遍历到第一个元素 2 时,候选元素变为 2,计数器变为 1。遍历到第二个元素 2 时,计数器变为 2。遍历到第三个元素 1 时,计数器变为 1。遍历到第四个元素 1 时,计数器变为 0,候选元素变为 1。遍历到第五个元素 1 时,计数器变为 1。遍历到第六个元素 2 时,计数器变为 0,候选元素变为 2。遍历到最后一个元素 2 时,计数器变为 1。最终的候选元素为 2,即为多数元素。

摩尔投票算法的时间复杂度为 O(n),空间复杂度为 O(1),是一种非常高效的解决多数元素问题的方法。

你可能感兴趣的:(蓝桥杯刷题,算法)